Home
last modified time | relevance | path

Searched refs:new_axis_mask (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc88 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceOp()
106 ellipsis_mask, new_axis_mask, shrink_axis_mask, in Compute()
151 final_shape.dims() == 2 && new_axis_mask == 0) { in Compute()
185 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
196 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceGradOp()
232 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
279 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
290 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceAssignOp()
352 new_axis_mask, shrink_axis_mask, &processing_shape, in Compute()
400 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc115 const int32 new_axis_mask = 0; in BM_ValidateStridedSliceOp() local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc38 const int32 new_axis_mask, shrink_axis_mask; member
112 } else if ((1 << i) & sparse.new_axis_mask) { in BuildDenseSpec()
156 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp() argument
207 new_axis_mask, in ValidateStridedSliceOp()
211 if (ellipsis_seen && ((1 << i) & new_axis_mask) != 0) { in ValidateStridedSliceOp()
379 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
387 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.cc569 int32 new_axis_mask, in StridedSliceGrad() argument
586 {"new_axis_mask", new_axis_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
629 {"new_axis_mask", new_axis_mask}, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
668 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
680 new_axis_mask, shrink_axis_mask); in TEST()
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
695 ellipsis_mask, new_axis_mask, shrink_axis_mask); in TEST()
707 new_axis_mask, shrink_axis_mask); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc437 uint64_t new_axis_mask = strided_slice_op.new_axis_mask().getZExtValue(); in matchAndRewrite() local
438 if (new_axis_mask == 0) return matchFailure(); in matchAndRewrite()
448 while (index < original_input_shape.size() || new_axis_mask) { in matchAndRewrite()
449 if (new_axis_mask & 1) { in matchAndRewrite()
454 new_axis_mask >>= 1; in matchAndRewrite()
478 new_begin_mask |= strided_slice_op.new_axis_mask(); in matchAndRewrite()
479 new_end_mask |= strided_slice_op.new_axis_mask(); in matchAndRewrite()
Dlegalize_tf.cc298 tf_strided_slice_op.new_axis_mask().getSExtValue()), in matchAndRewrite()
328 tf_strided_slice_op.new_axis_mask().getSExtValue()), in matchAndRewrite()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h60 p->new_axis_mask <<= pad_count; in StridedSlicePadIndices()
188 op_params.new_axis_mask = 0; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt58 name: "new_axis_mask"
92 produced using `new_axis_mask`. For example, `foo[:, ...]` where
129 new_axis_mask = 1<<2 4
146 end and stride, while the new_axis_mask bit is set.
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStridedSlice.pbtxt59 name: "new_axis_mask"
DResourceStridedSliceAssign.pbtxt59 name: "new_axis_mask"
DTensorStridedSliceUpdate.pbtxt63 name: "new_axis_mask"
DStridedSliceGrad.pbtxt63 name: "new_axis_mask"
DStridedSliceAssign.pbtxt65 name: "new_axis_mask"
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc77 op_params.new_axis_mask = 0; in BuildStridedSliceParams()
151 TF_LITE_ENSURE_MSG(context, op_context.params->new_axis_mask == 0, in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice_test.cc56 int end_mask, int ellipsis_mask, int new_axis_mask, in TestStrideSlide() argument
94 new_axis_mask, shrink_axis_mask}; in TestStrideSlide()
Dstrided_slice.cc80 op_params.new_axis_mask = 0; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1112 int new_axis_mask;
1127 new_axis_mask = other.new_axis_mask;
1161 new_axis_mask =
1162 toco::port::ReverseBits32(static_cast<uint32>(new_axis_mask)) >>
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc416 int64 new_axis_mask; in StridedSliceGradHelper() local
424 GetNodeAttr(op.node()->attrs(), "new_axis_mask", &new_axis_mask)); in StridedSliceGradHelper()
432 .NewAxisMask(new_axis_mask) in StridedSliceGradHelper()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py908 new_axis_mask, shrink_axis_mask = 0, 0
939 new_axis_mask |= (1 << index)
976 new_axis_mask=new_axis_mask,
1044 new_axis_mask=0, argument
1144 new_axis_mask=new_axis_mask,
1169 new_axis_mask=new_axis_mask,
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h345 int new_axis_mask; member
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc579 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
586 GetNodeAttr(slice_node->attrs(), "new_axis_mask", &new_axis_mask)); in PartialStridedSliceShape()
594 new_axis_mask != 0 || shrink_axis_mask != 0) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc38 CHECK_EQ(op.new_axis_mask, 0); in StridedSlice()
/external/tensorflow/tensorflow/compiler/tests/
Dnary_ops_test.py243 new_axis_mask=0x1)
254 ssg_test2 = lambda x: array_ops.strided_slice_grad(*x, new_axis_mask=0x15)
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dprepare-tf.mlir429 …{begin_mask = 15 : i64, ellipsis_mask = 0 : i64, end_mask = 15 : i64, new_axis_mask = 0 : i64, shr…
434 …{begin_mask = 15 : i64, ellipsis_mask = 0 : i64, end_mask = 15 : i64, new_axis_mask = 0 : i64, shr…
441 …) {begin_mask = 6 : i64, ellipsis_mask = 0 : i64, end_mask = 6 : i64, new_axis_mask = 9 : i64, shr…
448 …{begin_mask = 15 : i64, ellipsis_mask = 0 : i64, end_mask = 15 : i64, new_axis_mask = 0 : i64, shr…
456 … = 6 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 6 : i64, new_axis_mask = 1 : i64, shr…

123