| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 116 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
170 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
|
| D | types.h | 997 int16 shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 38 const int32 new_axis_mask, shrink_axis_mask; member
66 int32 shrink_axis_mask; member
156 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp()
379 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 182 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
275 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
373 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 116 const int32 shrink_axis_mask = 0; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 570 int32 shrink_axis_mask) { in StridedSliceGrad()
611 int32 shrink_axis_mask) { in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
719 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1, in TEST() local
747 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1, in TEST() local
|
| D | array_ops.cc | 1591 shrink_axis_mask; in __anon7c94107b2502() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 204 int shrink_axis_mask = op_context.params->shrink_axis_mask in Eval() local
|
| D | strided_slice_test.cc | 35 int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 417 int64 shrink_axis_mask; in StridedSliceGradHelper() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 1345 shrink_axis_mask): argument
|
| D | array_ops.py | 735 shrink_axis_mask=0, argument
|
| D | variables.py | 2123 shrink_axis_mask): argument
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 311 int shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 952 int16 shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3230 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3278 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1687 int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3146 const int shrink_axis_mask = node->attr().at("shrink_axis_mask").i(); in GetSliceAxis() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2618 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 846 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice()
|