Searched refs:shrink_axis_mask_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 41 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceOp()
72 shrink_axis_mask_, &dummy_processing_shape, &final_shape, in Compile()
107 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceOp
124 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceGradOp()
155 shrink_axis_mask_, &processing_shape, &final_shape, &dummy, in Compile()
216 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceGradOp
234 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceAssignOp()
270 shrink_axis_mask_, &dummy_processing_shape, &final_shape, in Compile()
323 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceAssignOp
|