/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | strided_slice_logic.h | 77 const auto begin_mask = params.begin_mask; in StartForAxis() local 183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
|
D | types.h | 1129 int16_t begin_mask; member
|
/external/tensorflow/tensorflow/core/kernels/ |
D | strided_slice_op.cc | 184 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp 278 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp 390 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
|
D | strided_slice_op_test.cc | 109 const int32 begin_mask = 0; in BM_ValidateStridedSliceOp() local
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_grad_test.cc | 567 const Tensor& dy, int32 begin_mask, in StridedSliceGrad() 610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_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 | 1729 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anondb9326b22502() local
|
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
D | strided_slice_builder.cc | 51 int begin_mask = params->begin_mask; in PopulateSubGraph() local
|
/external/tensorflow/tensorflow/core/util/ |
D | strided_slice_op.cc | 38 const int32 begin_mask, end_mask; member 50 int32 begin_mask; member
|
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
D | strided_slice_test.cc | 30 std::initializer_list<int> strides_data, int begin_mask, in StridedSliceOpModel()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | strided_slice_op.cc | 142 bool begin_mask = (1 << i) & shape_spec.begin_dense_mask; in EmitDynamicSlice() local 443 bool begin_mask = (1 << i) & shape_spec.begin_dense_mask; in CompileAsDynamicUpdateSlice() local
|
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
D | lstm_utils.cc | 164 int64_t begin_mask, int64_t end_mask, in CreateStridedSliceOp() 779 int64_t begin_mask; in ConvertKerasLSTMLayer() local
|
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
D | types.h | 133 int16_t begin_mask; member
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | prepare_tf.cc | 645 int64_t begin_mask = strided_slice_op.begin_mask(); in RewriteEllipsisMask() local 799 int begin_mask = strided_slice_op.begin_mask(); in matchAndRewrite() local
|
/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
|
D | shape_refiner_test.cc | 66 int begin_mask = 0, int end_mask = 0, in TestStridedSlice()
|
/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 1761 MutableArrayRef<int64_t> input_shape, int32_t begin_mask, int32_t end_mask, in CalculateSlicedShapeFromDenseIndices() 1827 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in CalculateSlicedShapeFromSparseIndices()
|
/external/tensorflow/tensorflow/cc/gradients/ |
D | array_grad.cc | 422 int64 begin_mask; in StridedSliceGradHelper() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | strided_slice_test.cc | 36 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel()
|
/external/tensorflow/tensorflow/lite/c/ |
D | builtin_op_data.h | 394 int begin_mask; member
|
/external/tensorflow/tensorflow/python/ops/ |
D | resource_variable_ops.py | 1357 def _strided_slice_assign(self, begin, end, strides, value, name, begin_mask, argument
|
D | array_ops.py | 1117 begin_mask=0, argument
|
D | variables.py | 2541 def _strided_slice_assign(self, begin, end, strides, value, name, begin_mask, argument
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 3219 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local 3267 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
D | legalize_common.cc | 2002 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in convertStridedSliceOp()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 1827 int begin_mask = 0; in MaybeUpdateNodeContextOutput() local
|