Home
last modified time | relevance | path

Searched refs:shrink_axis_mask (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc38 const int32 new_axis_mask, shrink_axis_mask; member
66 int32 shrink_axis_mask; member
82 dense->shrink_axis_mask = 0; in BuildDenseSpec()
134 if (sparse.shrink_axis_mask & (1 << i)) { in BuildDenseSpec()
136 dense->shrink_axis_mask |= (1 << full_index); in BuildDenseSpec()
151 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp() argument
203 shrink_axis_mask}; in ValidateStridedSliceOp()
259 bool shrink_i = (dense_spec.shrink_axis_mask & (1 << i)); in ValidateStridedSliceOp()
374 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
382 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask, in ValidateStridedSliceOp()
Dstrided_slice_op.h47 int32 new_axis_mask, int32 shrink_axis_mask,
58 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc522 int32 shrink_axis_mask) { in StridedSliceGrad() argument
539 {"shrink_axis_mask", shrink_axis_mask}, in StridedSliceGrad()
563 int32 shrink_axis_mask) { in StridedSliceGradGrad() argument
582 {"shrink_axis_mask", shrink_axis_mask}, in StridedSliceGradGrad()
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
620 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
632 new_axis_mask, shrink_axis_mask); in TEST()
643 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
647 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
659 new_axis_mask, shrink_axis_mask); in TEST()
[all …]
Darray_ops.cc1454 shrink_axis_mask; in __anon0d47aeef2602() local
1459 TF_RETURN_IF_ERROR(c->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in __anon0d47aeef2602()
1469 end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask, in __anon0d47aeef2602()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc87 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceOp()
103 ellipsis_mask, new_axis_mask, shrink_axis_mask, in Compute()
174 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
187 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradOp()
222 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
266 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
279 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceAssignOp()
312 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
358 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc116 const int32 shrink_axis_mask = 0; in BM_ValidateStridedSliceOp() local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dstrided_slice.cc125 if (!(op_context->params->shrink_axis_mask & (1 << idx))) { in ResizeOutputTensor()
206 op_context.params->shrink_axis_mask = in Eval()
207 ReverseMaskBits(op_context.params->shrink_axis_mask, op_context.dims); in Eval()
213 op_context.params->end_mask, op_context.params->shrink_axis_mask, \ in Eval()
Dstrided_slice_test.cc34 int shrink_axis_mask) { in StridedSliceOpModel() argument
43 new_axis_mask, shrink_axis_mask) in StridedSliceOpModel()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt66 name: "shrink_axis_mask"
72 `shrink_axis_mask` being 2.
115 `shrink_axis_mask`.
130 shrink_axis_mask = 1<<0
139 also set the appropriate bit in `shrink_axis_mask`.
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py521 new_axis_mask, shrink_axis_mask = 0, 0
557 shrink_axis_mask |= (1 << index)
585 shrink_axis_mask=shrink_axis_mask,
655 shrink_axis_mask=0, argument
755 shrink_axis_mask=shrink_axis_mask)
778 shrink_axis_mask=shrink_axis_mask)
Dresource_variable_ops.py826 shrink_axis_mask): argument
839 shrink_axis_mask=shrink_axis_mask))
Dvariables.py672 shrink_axis_mask): argument
683 shrink_axis_mask=shrink_axis_mask)
Darray_grad.py275 shrink_axis_mask=op.get_attr("shrink_axis_mask")), None, None, None
294 shrink_axis_mask=op.get_attr("shrink_axis_mask"))
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h214 int shrink_axis_mask; member
Dmodel.cc564 params->shrink_axis_mask = schema_params->shrink_axis_mask(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc368 op.shrink_axis_mask = 2; in TEST_F()
380 EXPECT_EQ(op.shrink_axis_mask, output_toco_op->shrink_axis_mask); in TEST_F()
Doperator.cc633 op.new_axis_mask, op.shrink_axis_mask); in WriteOptions()
642 op->shrink_axis_mask = options.shrink_axis_mask(); in ReadOptions()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py894 shrink_axis_mask = tensor.op.get_attr("shrink_axis_mask")
896 not shrink_axis_mask and (not begin_mask or
/external/tensorflow/tensorflow/compiler/tests/
Dnary_ops_test.py242 return array_ops.strided_slice_grad(*x, shrink_axis_mask=0x4,
Drandomized_tests.cc3096 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3109 .Attr("shrink_axis_mask", shrink_axis_mask)); in TEST_F()
3144 int64 shrink_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3162 .Attr("shrink_axis_mask", shrink_axis_mask)); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/schema/
Dschema.fbs364 shrink_axis_mask: int;
Dschema_generated.h3475 int32_t shrink_axis_mask;
3481 shrink_axis_mask(0) {
3506 int32_t shrink_axis_mask() const {
3538 void add_shrink_axis_mask(int32_t shrink_axis_mask) {
3539 fbb_.AddElement<int32_t>(StridedSliceOptions::VT_SHRINK_AXIS_MASK, shrink_axis_mask, 0);
3559 int32_t shrink_axis_mask = 0) {
3561 builder_.add_shrink_axis_mask(shrink_axis_mask);
5393 { auto _e = shrink_axis_mask(); _o->shrink_axis_mask = _e; };
5408 auto _shrink_axis_mask = _o->shrink_axis_mask;
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h811 int shrink_axis_mask;
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Dreference_ops.h2700 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice() argument
2712 end_mask & 8, shrink_axis_mask & 8); in StridedSlice()
2717 end_mask & 4, shrink_axis_mask & 4); in StridedSlice()
2722 end_mask & 2, shrink_axis_mask & 2); in StridedSlice()
2727 end_mask & 1, shrink_axis_mask & 1); in StridedSlice()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1202 if (op->shrink_axis_mask & mask) { in ProcessStridedSliceOperator()

12