| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 126 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
|
| D | types.h | 1133 int16_t shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 40 const int32 new_axis_mask, shrink_axis_mask; member
74 int32 shrink_axis_mask; member
173 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp()
421 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 185 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
279 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
391 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 113 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 | 1730 shrink_axis_mask; in __anon42d741192502() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | strided_slice_builder.cc | 53 int shrink_axis_mask = params->shrink_axis_mask; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | strided_slice_test.cc | 31 int end_mask, int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 166 int64_t shrink_axis_mask, OpBuilder* builder) { in CreateStridedSliceOp()
781 int64_t shrink_axis_mask; in ConvertKerasLSTMLayer() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 137 int16_t shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 586 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1676 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; member
1687 int32_t begin_mask, end_mask, shrink_axis_mask; member
1762 int32_t shrink_axis_mask, MutableArrayRef<int64_t> begin, in CalculateSlicedShapeFromDenseIndices()
1828 int32_t new_axis_mask, int32_t shrink_axis_mask, in CalculateSlicedShapeFromSparseIndices()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 426 int64 shrink_axis_mask; in StridedSliceGradHelper() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 38 int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 598 uint64_t shrink_axis_mask = strided_slice_op.shrink_axis_mask(); in RewriteEllipsisMask() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 398 int shrink_axis_mask; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 1359 shrink_axis_mask): argument
|
| D | array_ops.py | 1121 shrink_axis_mask=0, argument
|
| D | variables.py | 2543 shrink_axis_mask): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3231 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3279 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2003 int32_t new_axis_mask, int32_t shrink_axis_mask) { in convertStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3602 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F()
3732 int shrink_axis_mask; in TEST_F() member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3330 const int shrink_axis_mask = node->attr().at("shrink_axis_mask").i(); in GetStridedSliceAxis() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2518 int shrink_axis_mask) { in StridedSliceOpModel()
|