Searched refs:ellipsis_mask_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 39 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceOp()
71 begin_mask_, end_mask_, ellipsis_mask_, new_axis_mask_, in Compile()
107 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceOp
122 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceGradOp()
154 begin_mask_, end_mask_, ellipsis_mask_, new_axis_mask_, in Compile()
216 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceGradOp
232 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceAssignOp()
269 begin_mask_, end_mask_, ellipsis_mask_, new_axis_mask_, in Compile()
323 int32 ellipsis_mask_, new_axis_mask_, shrink_axis_mask_; member in tensorflow::__anon2611573b0111::StridedSliceAssignOp
|