experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add bug number for TODO

Browse Source 
PiperOrigin-RevId: 347642991
Change-Id: Ic8535c2ebcf80e4ed265d0e99914c4507d915298
This commit is contained in:
T.J. Alumbaugh 2020-12-15 10:30:32 -08:00 committed by TensorFlower Gardener
parent ddfc8bd0fb
commit c6228dd44a
1 changed files with 2 additions and 3 deletions

5
tensorflow/lite/kernels/strided_slice.cc
View File

@ -37,7 +37,7 @@ namespace strided_slice {
enum KernelType {
kReference,
// TODO(soroosh): add kGenericOptimized
// TODO(b/175642009): add kGenericOptimized
};
constexpr int kInputTensor = 0;
@ -154,7 +154,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.strides), 1);
TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type);
// Only INT32 begin/end/strides are supported
// TODO(soroosh) add support for INT64
// TODO(b/175642009): add support for INT64
TF_LITE_ENSURE_TYPES_EQ(context, op_context.begin->type, kTfLiteInt32);
TF_LITE_ENSURE_TYPES_EQ(context, op_context.end->type, kTfLiteInt32);
TF_LITE_ENSURE_TYPES_EQ(context, op_context.strides->type, kTfLiteInt32);
@ -256,7 +256,6 @@ TfLiteRegistration* Register_STRIDED_SLICE_REF() {
return &r;
}
// TODO(soroosh): add optimized
TfLiteRegistration* Register_STRIDED_SLICE() {
return Register_STRIDED_SLICE_REF();
}