| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | strided_slice_test.cc | 32 std::initializer_list<int> strides_shape, int begin_mask,
42 CreateStridedSliceOptions(builder_, begin_mask, end_mask, ellipsis_mask,
29 StridedSliceOpModel(std::initializer_list<int> input_shape, std::initializer_list<int> begin_shape, std::initializer_list<int> end_shape, std::initializer_list<int> strides_shape, int begin_mask, int end_mask, int ellipsis_mask, int new_axis_mask, int shrink_axis_mask) argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op_test.cc | 112 const int32 begin_mask = 0; local
119 &begin, &end, strides, input_shape, begin_mask, end_mask, ellipsis_mask,
|
| H A D | strided_slice_op.cc | 82 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask));
102 context->input(0).shape(), begin_mask, end_mask,
173 int32 begin_mask, end_mask; member in class:tensorflow::StridedSliceOp
182 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask));
221 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask,
265 int32 begin_mask, end_mask; member in class:tensorflow::StridedSliceGradOp
274 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask));
357 int32 begin_mask, end_mask; member in class:tensorflow::StridedSliceAssignOp
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| H A D | array_grad_test.cc | 519 const Tensor& dy, int32 begin_mask,
536 {"begin_mask", begin_mask},
562 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask,
579 {"begin_mask", begin_mask},
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, local
619 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask,
631 begin_mask, end_mask, ellipsis_mask,
643 int begin_mask local
517 StridedSliceGrad(const Tensor& x, const Tensor& begin, const Tensor& end, const Tensor& strides, const Tensor& dy, int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, int32 shrink_axis_mask) argument
559 StridedSliceGradGrad( const Tensor& shape, const Tensor& begin, const Tensor& end, const Tensor& strides, const Tensor& dy, const Tensor& grad, int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, int32 shrink_axis_mask) argument
671 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1, local
699 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1, local
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 36 const int32 begin_mask, end_mask; member in struct:tensorflow::__anon26833::StridedSliceSparseSpec
48 int32 begin_mask; member in struct:tensorflow::__anon26833::StridedSliceDenseSpec
74 // Build expanded begin, end, strides, begin_mask, end_mask
80 dense->begin_mask = 0;
101 dense->begin_mask |= (1 << full_index);
125 if (sparse.begin_mask & (1 << i)) {
126 dense->begin_mask |= (1 << full_index);
225 // we need to produce the missing begin_mask for the first two
227 // we achieve begin_mask=6, end_mask=7
229 0 /* begin_mask */,
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/ |
| H A D | builtin_op_data.h | 210 int begin_mask; member in struct:__anon25898
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| H A D | model.h | 807 int begin_mask; member in struct:toco::StridedSliceOperator
|
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| H A D | schema_generated.h | 3471 int32_t begin_mask; member in struct:tflite::StridedSliceOptionsT
3477 : begin_mask(0),
3494 int32_t begin_mask() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
3526 void add_begin_mask(int32_t begin_mask) { argument
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;
[all...] |