Searched refs:ConstantInputAsInt64Literal (Results 1 – 10 of 10) 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 | mirror_pad_op.cc | 92 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); in Compile()
191 ctx->ConstantInputAsInt64Literal("paddings", &pad_literal)); 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 | reduce_window_op.cc | 110 OP_REQUIRES_OK(context, context->ConstantInputAsInt64Literal( in Compile()
|
| D | xla_select_and_scatter_op.cc | 117 OP_REQUIRES_OK(context, context->ConstantInputAsInt64Literal( in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 150 Status ConstantInputAsInt64Literal(int index, xla::Literal* out);
151 Status ConstantInputAsInt64Literal(absl::string_view name, xla::Literal* out);
|
| D | xla_op_kernel.cc | 372 Status XlaOpKernelContext::ConstantInputAsInt64Literal(int index, in ConstantInputAsInt64Literal() function in tensorflow::XlaOpKernelContext
397 Status XlaOpKernelContext::ConstantInputAsInt64Literal(absl::string_view name, in ConstantInputAsInt64Literal() function in tensorflow::XlaOpKernelContext
400 return ConstantInputAsInt64Literal(index, out); in ConstantInputAsInt64Literal()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 112 ctx->ConstantInputAsInt64Literal(1, &source_target_literal)); in Compile()
|