Searched defs:begin_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 59 void TestStridedSliceFloat(const int* input_shape, const int* begin_shape, in TestStridedSliceFloat()
91 const int* input_shape, const int* begin_shape, const int* end_shape, in TestStridedSliceQuantized()
128 const int begin_shape[] = {1, 5}; in TF_LITE_MICRO_TEST() local
149 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
170 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
191 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
212 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
233 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
254 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
275 const int begin_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | slice_test.cc | 40 std::initializer_list<int> begin_shape, in SliceOpModel()
|
| D | strided_slice_test.cc | 34 std::initializer_list<int> begin_shape, in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 199 const TensorShape begin_shape = ctx->InputShape("begin"); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 129 ArrayRef<int64_t> begin_shape, in SliceRankedTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 638 const ArrayRef<int64_t> begin_shape = begin_type.getShape(); in RewriteEllipsisMask() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1706 ShapeHandle begin_shape, end_shape, strides_shape; in __anon42d741192502() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2153 ShapeHandle begin_shape; in SliceShape() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 972 TensorShape begin_shape({shape.dims()}); in CreateSliceNode() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2511 std::initializer_list<int> begin_shape, in StridedSliceOpModel()
|