Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc36 const int32 begin_mask, end_mask; member
48 int32 begin_mask; member
80 dense->begin_mask = 0; in BuildDenseSpec()
101 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
125 if (sparse.begin_mask & (1 << i)) { in BuildDenseSpec()
126 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
266 {dense_spec.begin_mask & (1 << i), dense_spec.end_mask & (1 << i)}}; in ValidateStridedSliceOp()
287 (dense_spec.begin_mask & (1 << i)) && (dense_spec.end_mask & (1 << i)); in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc519 const Tensor& dy, int32 begin_mask, in StridedSliceGrad() argument
536 {"begin_mask", begin_mask}, in StridedSliceGrad()
562 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
579 {"begin_mask", begin_mask}, in StridedSliceGradGrad()
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
619 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
631 begin_mask, end_mask, ellipsis_mask, in TEST()
643 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
646 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
658 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
Darray_ops.cc1453 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon0d47aeef2602() local
1455 TF_RETURN_IF_ERROR(c->GetAttr("begin_mask", &begin_mask)); in __anon0d47aeef2602()
1468 begin_value, end_value, *strides_value, input_shape, begin_mask, in __anon0d47aeef2602()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc82 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
102 context->input(0).shape(), begin_mask, end_mask, in Compute()
173 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
182 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
221 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
265 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
274 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceAssignOp()
311 old_lhs.shape(), begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
357 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc112 const int32 begin_mask = 0; in BM_ValidateStridedSliceOp() local
119 &begin, &end, strides, input_shape, begin_mask, end_mask, ellipsis_mask, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt9 `begin_mask` then `begin[k]` is ignored and the full range of the
31 name: "begin_mask"
42 analogous to `begin_mask`
102 the second to last. `begin_mask` controls whether to replace the explicitly
126 begin_mask = 1<<4 | 1 << 5 = 48
155 The appropriate begin_mask bit is set to indicate the start range is the
160 receive 0, 0, and 1, respectively. The appropriate bits in `begin_mask` and
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dstrided_slice.cc93 return op_context->params->begin_mask & (1 << idx) in GetBeginValueAtIndex()
202 op_context.params->begin_mask = in Eval()
203 ReverseMaskBits(op_context.params->begin_mask, op_context.dims); in Eval()
212 GetTensorDims(op_context.input), op_context.params->begin_mask, \ in Eval()
Dstrided_slice_test.cc32 std::initializer_list<int> strides_shape, int begin_mask, in StridedSliceOpModel() argument
42 CreateStridedSliceOptions(builder_, begin_mask, end_mask, ellipsis_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py522 begin_mask, end_mask = 0, 0
533 begin_mask |= (1 << index)
583 begin_mask=begin_mask,
651 begin_mask=0, argument
751 begin_mask=begin_mask,
774 begin_mask=begin_mask,
Dresource_variable_ops.py824 def _strided_slice_assign(self, begin, end, strides, value, name, begin_mask, argument
835 begin_mask=begin_mask,
Dvariables.py668 begin_mask, argument
679 begin_mask=begin_mask,
Darray_grad.py271 begin_mask=op.get_attr("begin_mask"),
290 begin_mask=op.get_attr("begin_mask"),
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h210 int begin_mask; member
Dmodel.cc560 params->begin_mask = schema_params->begin_mask(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc364 op.begin_mask = 1; in TEST_F()
375 EXPECT_EQ(op.begin_mask, output_toco_op->begin_mask); in TEST_F()
Doperator.cc632 *builder, op.begin_mask, op.end_mask, op.ellipsis_mask, in WriteOptions()
638 op->begin_mask = options.begin_mask(); in ReadOptions()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py885 begin_mask = tensor.op.get_attr("begin_mask")
886 if begin_mask == 1:
896 not shrink_axis_mask and (not begin_mask or
897 (begin_mask == 1)) and
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc29 if (op.begin_mask & 1 << axis) { in StartForAxis()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Dreference_ops.h2700 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice() argument
2709 StartIndex(starts[3], strides[3], input_dims.sizes[3], begin_mask & 8); in StridedSlice()
2714 StartIndex(starts[2], strides[2], input_dims.sizes[2], begin_mask & 4); in StridedSlice()
2719 StartIndex(starts[1], strides[1], input_dims.sizes[1], begin_mask & 2); in StridedSlice()
2724 StartIndex(starts[0], strides[0], input_dims.sizes[0], begin_mask & 1); in StridedSlice()
2747 int begin_mask, int end_mask, in StridedSlice() argument
2752 StridedSlice(input_data, input_dims, begin_mask, end_mask, in StridedSlice()
/external/tensorflow/tensorflow/contrib/lite/schema/
Dschema.fbs360 begin_mask: int;
Dschema_generated.h3471 int32_t begin_mask;
3477 : begin_mask(0),
3494 int32_t begin_mask() const {
3526 void add_begin_mask(int32_t begin_mask) {
3527 fbb_.AddElement<int32_t>(StridedSliceOptions::VT_BEGIN_MASK, begin_mask, 0);
3555 int32_t begin_mask = 0,
3565 builder_.add_begin_mask(begin_mask);
5389 { auto _e = begin_mask(); _o->begin_mask = _e; };
5404 auto _begin_mask = _o->begin_mask;
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h807 int begin_mask;
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3084 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local
3105 .Attr("begin_mask", begin_mask) in TEST_F()
3132 int64 begin_mask = bitmask_distribution(generator()); in TEST_F() local
3158 .Attr("begin_mask", begin_mask) in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4416 int begin_mask, int end_mask, in StridedSlice() argument
4422 const int start_b = (begin_mask & 8) ? 0 : starts[3]; in StridedSlice()
4424 const int start_h = (begin_mask & 4) ? 0 : starts[2]; in StridedSlice()
4426 const int start_w = (begin_mask & 2) ? 0 : starts[1]; in StridedSlice()
4428 const int start_d = (begin_mask & 1) ? 0 : starts[0]; in StridedSlice()
/external/tensorflow/tensorflow/contrib/lite/testing/
Dgenerate_examples.py1707 begin_mask=parameters["begin_mask"],

12