Searched refs:ConstantInputAsInt64Literal (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 155 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsInt64Literal(2, &crops)); in Compile()
177 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsInt64Literal(1, &crops)); in Compile()
|
| D | spacetobatch_op.cc | 157 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsInt64Literal(2, &paddings)); in Compile()
179 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsInt64Literal(1, &paddings)); in Compile()
|
| D | pad_op.cc | 56 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); in Compile()
|
| D | xla_conv_op.cc | 67 OP_REQUIRES_OK(context, context->ConstantInputAsInt64Literal( in Compile()
|
| D | mirror_pad_op.cc | 90 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); in Compile()
|
| D | reduce_window_op.cc | 111 OP_REQUIRES_OK(context, context->ConstantInputAsInt64Literal( in Compile()
|
| D | xla_select_and_scatter_op.cc | 118 OP_REQUIRES_OK(context, context->ConstantInputAsInt64Literal( in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 135 Status ConstantInputAsInt64Literal(int index, xla::Literal* out);
136 Status ConstantInputAsInt64Literal(absl::string_view name, xla::Literal* out);
|
| D | xla_op_kernel.cc | 280 Status XlaOpKernelContext::ConstantInputAsInt64Literal(int index, in ConstantInputAsInt64Literal() function in tensorflow::XlaOpKernelContext
305 Status XlaOpKernelContext::ConstantInputAsInt64Literal(absl::string_view name, in ConstantInputAsInt64Literal() function in tensorflow::XlaOpKernelContext
308 return ConstantInputAsInt64Literal(index, out); in ConstantInputAsInt64Literal()
|