| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 37 int32 ellipsis_mask; member
98 if ((1 << i) & sparse.ellipsis_mask) { in BuildDenseSpec()
155 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp() argument
187 if (ellipsis_mask && ((ellipsis_mask & (ellipsis_mask - 1)) != 0)) { in ValidateStridedSliceOp()
206 ellipsis_mask, in ValidateStridedSliceOp()
214 if ((1 << i) & ellipsis_mask) { in ValidateStridedSliceOp()
220 sparse_spec.ellipsis_mask |= (1 << sparse_spec.dims); in ValidateStridedSliceOp()
378 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp() argument
387 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask, in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 46 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask,
57 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 568 int32 end_mask, int32 ellipsis_mask, in StridedSliceGrad() argument
588 {"ellipsis_mask", ellipsis_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
631 {"ellipsis_mask", ellipsis_mask}, in StridedSliceGradGrad()
665 ellipsis_mask = 0; in TEST() local
668 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
679 begin_mask, end_mask, ellipsis_mask, in TEST()
692 ellipsis_mask = 0; in TEST() local
695 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
706 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 85 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceOp()
104 ellipsis_mask, new_axis_mask, shrink_axis_mask, in Compute()
182 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
192 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceGradOp()
229 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
275 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
285 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceAssignOp()
325 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
373 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 114 const int32 ellipsis_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/ |
| D | api_def_StridedSlice.pbtxt | 46 name: "ellipsis_mask"
50 If `ellipsis_mask == 0`, then an implicit ellipsis mask of `1 << (m+1)`
89 `ellipsis_mask`. For example, `foo[...]` is the identity slice.
128 ellipsis_mask = 1<<3 = 8
165 `ellipsis_mask must be a power of two (only one ellipsis)`
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 59 p->ellipsis_mask <<= pad_count; in StridedSlicePadIndices()
186 op_params.ellipsis_mask = 0; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 2590 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
2715 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
2897 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3042 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3276 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3421 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3714 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3856 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
3998 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
4535 key: "ellipsis_mask" in CreateImagenetDatasetFunctions()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 599 ellipsis_mask = 0
626 ellipsis_mask |= (1 << index)
667 ellipsis_mask=ellipsis_mask,
733 ellipsis_mask=0, argument
833 ellipsis_mask=ellipsis_mask,
857 ellipsis_mask=ellipsis_mask,
|
| D | resource_variable_ops.py | 1344 end_mask, ellipsis_mask, new_axis_mask, argument
1357 ellipsis_mask=ellipsis_mask,
|
| D | variables.py | 2121 ellipsis_mask, argument
2132 ellipsis_mask=ellipsis_mask,
|
| D | array_grad.py | 275 ellipsis_mask=op.get_attr("ellipsis_mask"),
294 ellipsis_mask=op.get_attr("ellipsis_mask"),
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 1063 int ellipsis_mask;
1078 ellipsis_mask = other.ellipsis_mask;
1109 ellipsis_mask =
1110 toco::port::ReverseBits32(static_cast<uint32>(ellipsis_mask)) >>
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 415 int64 ellipsis_mask; in StridedSliceGradHelper() local
422 GetNodeAttr(op.node()->attrs(), "ellipsis_mask", &ellipsis_mask)); in StridedSliceGradHelper()
431 .EllipsisMask(ellipsis_mask) in StridedSliceGradHelper()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 309 int ellipsis_mask; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
566 GetNodeAttr(slice_node->attrs(), "ellipsis_mask", &ellipsis_mask)); in PartialStridedSliceShape()
575 !(end_mask == 0 || end_mask == 1) || ellipsis_mask != 0 || in PartialStridedSliceShape()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 37 CHECK_EQ(op.ellipsis_mask, 0); in StridedSlice()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 34 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel() argument
43 CreateStridedSliceOptions(builder_, begin_mask, end_mask, ellipsis_mask, in StridedSliceOpModel()
|
| D | strided_slice.cc | 162 TF_LITE_ENSURE_MSG(context, op_context.params->ellipsis_mask == 0, in Prepare()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 543 params->ellipsis_mask = schema_params->ellipsis_mask(); in ParseOpData()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1851 ellipsis_mask = pfor_input.get_attr("ellipsis_mask")
1860 ellipsis_mask <<= 1
1871 ellipsis_mask=ellipsis_mask,
1885 ellipsis_mask = pfor_input.get_attr("ellipsis_mask")
1895 ellipsis_mask <<= 1
1907 ellipsis_mask=ellipsis_mask,
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 459 op.ellipsis_mask = 1; in TEST_F()
471 EXPECT_EQ(op.ellipsis_mask, output_toco_op->ellipsis_mask); in TEST_F()
|
| D | operator.cc | 1485 *builder, op.begin_mask, op.end_mask, op.ellipsis_mask, in WriteOptions()
1493 op->ellipsis_mask = options.ellipsis_mask(); in ReadOptions()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 914 ellipsis_mask = tensor.op.get_attr("ellipsis_mask")
917 valid_attributes = (not ellipsis_mask and not new_axis_mask and
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3222 int64 ellipsis_mask = 0; in TEST_F() local
3226 ellipsis_mask = 1LL << ellipsis_pos; in TEST_F()
3241 .Attr("ellipsis_mask", ellipsis_mask) in TEST_F()
3270 int64 ellipsis_mask = 0; in TEST_F() local
3274 ellipsis_mask = 1LL << ellipsis_pos; in TEST_F()
3294 .Attr("ellipsis_mask", ellipsis_mask) in TEST_F()
|