Searched refs:strides_literal (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 53 xla::Literal begin_literal, end_literal, strides_literal; in Compile() local
56 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
63 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
137 xla::Literal begin_literal, end_literal, strides_literal; in Compile() local
140 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
147 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
245 xla::Literal begin_literal, end_literal, strides_literal; in Compile() local
248 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
255 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
|