Home
last modified time | relevance | path

Searched defs:begin_mask (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h76 const auto begin_mask = params.begin_mask; in StartForAxis() local
170 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
Dtypes.h993 int16 begin_mask; member
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc181 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
274 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
372 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc112 const int32 begin_mask = 0; in BM_ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc567 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
Darray_ops.cc1590 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon7c94107b2502() local
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc36 const int32 begin_mask, end_mask; member
48 int32 begin_mask; member
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc202 int begin_mask = op_context.params->begin_mask << (4 - op_context.dims); in Eval() local
Dstrided_slice_test.cc33 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
Dshape_refiner_test.cc66 int begin_mask = 0, int end_mask = 0, in TestStridedSlice()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc413 int64 begin_mask; in StridedSliceGradHelper() local
/external/tensorflow/tensorflow/python/ops/
Dresource_variable_ops.py1343 def _strided_slice_assign(self, begin, end, strides, value, name, begin_mask, argument
Darray_ops.py731 begin_mask=0, argument
Dvariables.py2119 begin_mask, argument
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h307 int begin_mask; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2618 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F()
2759 int begin_mask; in TEST_F() member
Dconvert_nodes.cc2573 const int begin_mask = attrs.get<int64>("begin_mask"); in ConvertStridedSlice() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1527 int begin_mask = 0; in MaybeUpdateNodeContextOutput() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h948 int16 begin_mask; member
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3218 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local
3266 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2141 int begin_mask = node->attr().at("begin_mask").i(); in SimplifyStridedSlice() local
Darithmetic_optimizer.cc3142 const int begin_mask = node->attr().at("begin_mask").i(); in GetSliceAxis() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1685 std::initializer_list<int> strides_data, int begin_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h846 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice()