| /external/tensorflow/tensorflow/core/util/ |
| strided_slice_op.h | 47 int32 new_axis_mask, int32 shrink_axis_mask,
58 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
|
| strided_slice_op.cc | 38 const int32 new_axis_mask, shrink_axis_mask; member in struct:tensorflow::__anon40154::StridedSliceSparseSpec
66 int32 shrink_axis_mask; member in struct:tensorflow::__anon40154::StridedSliceDenseSpec
82 dense->shrink_axis_mask = 0;
134 if (sparse.shrink_axis_mask & (1 << i)) {
136 dense->shrink_axis_mask |= (1 << full_index);
151 int32 new_axis_mask, int32 shrink_axis_mask,
203 shrink_axis_mask};
259 bool shrink_i = (dense_spec.shrink_axis_mask & (1 << i));
374 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
382 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask,
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_grad_test.cc | 522 int32 shrink_axis_mask) {
539 {"shrink_axis_mask", shrink_axis_mask},
563 int32 shrink_axis_mask) {
582 {"shrink_axis_mask", shrink_axis_mask},
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, local
620 ellipsis_mask, new_axis_mask, shrink_axis_mask);
632 new_axis_mask, shrink_axis_mask);
643 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0 local
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...] |
| array_ops.cc | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op.cc | 87 context->GetAttr("shrink_axis_mask", &shrink_axis_mask));
103 ellipsis_mask, new_axis_mask, shrink_axis_mask,
174 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in class:tensorflow::StridedSliceOp
187 context->GetAttr("shrink_axis_mask", &shrink_axis_mask));
222 shrink_axis_mask, &processing_shape, &final_shape, &is_identity,
266 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in class:tensorflow::StridedSliceGradOp
279 context->GetAttr("shrink_axis_mask", &shrink_axis_mask));
358 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in class:tensorflow::StridedSliceAssignOp
[all...] |
| strided_slice_op_test.cc | 116 const int32 shrink_axis_mask = 0; local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape,
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| builtin_op_data.h | 214 int shrink_axis_mask; member in struct:__anon39219
|
| model.cc | 564 params->shrink_axis_mask = schema_params->shrink_axis_mask();
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| strided_slice.cc | 125 if (!(op_context->params->shrink_axis_mask & (1 << idx))) {
206 op_context.params->shrink_axis_mask =
207 ReverseMaskBits(op_context.params->shrink_axis_mask, op_context.dims);
213 op_context.params->end_mask, op_context.params->shrink_axis_mask, \
|
| strided_slice_test.cc | 34 int shrink_axis_mask) {
43 new_axis_mask, shrink_axis_mask)
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_ops.py | 521 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,
678 `new_axis_mask`, `shrink_axis_mask`) the ith bit will correspond to
703 If the ith bit of `shrink_axis_mask` is set, it implies that the ith
707 `shrink_axis_mask` equal to 2.
734 shrink_axis_mask: An `int32` mask.
755 shrink_axis_mask=shrink_axis_mask
[all...] |
| resource_variable_ops.py | [all...] |
| variables.py | 672 shrink_axis_mask):
683 shrink_axis_mask=shrink_axis_mask)
[all...] |
| array_grad.py | 275 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/toco/tflite/ |
| operator_test.cc | 368 op.shrink_axis_mask = 2;
380 EXPECT_EQ(op.shrink_axis_mask, output_toco_op->shrink_axis_mask);
|
| operator.cc | 633 op.new_axis_mask, op.shrink_axis_mask);
642 op->shrink_axis_mask = options.shrink_axis_mask();
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| nary_ops_test.py | 242 return array_ops.strided_slice_grad(*x, shrink_axis_mask=0x4,
|
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| model.h | 811 int shrink_axis_mask; member in struct:toco::StridedSliceOperator
[all...] |
| export_tensorflow.cc | [all...] |
| import_tensorflow.cc | [all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| tensor_util.py | 894 shrink_axis_mask = tensor.op.get_attr("shrink_axis_mask"
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| schema_generated.h | 3475 int32_t shrink_axis_mask; member in struct:tflite::StridedSliceOptionsT
3506 int32_t shrink_axis_mask() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| propagate_fixed_sizes.cc | [all...] |
