Searched refs:ConstantInput (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
55 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
56 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
138 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
139 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
140 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
246 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
247 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
248 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
|
| D | sequence_ops.cc | 86 OP_REQUIRES_OK(ctx, ctx->ConstantInput(0, &start)); in Compile()
87 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &limit)); in Compile()
88 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &delta)); in Compile()
147 OP_REQUIRES_OK(ctx, ctx->ConstantInput("start", &start_literal)); in Compile()
149 OP_REQUIRES_OK(ctx, ctx->ConstantInput("stop", &stop_literal)); in Compile()
|
| D | reverse_op.cc | 54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &lax)); in Compile()
|
| D | stateful_random_ops.cc | 193 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in CompileImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 113 Status XlaOpKernelContext::ConstantInput(int index, in ConstantInput() function in tensorflow::XlaOpKernelContext
132 Status XlaOpKernelContext::ConstantInput(absl::string_view name, in ConstantInput() function in tensorflow::XlaOpKernelContext
135 return ConstantInput(index, constant_literal); in ConstantInput()
212 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsIntScalar()
224 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsFloatScalar()
253 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsIntVector()
283 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsInt64Literal()
315 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsShape()
325 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsPartialShape()
363 TF_RETURN_IF_ERROR(ConstantInput(i, &(*outputs)[i])); in ConstantInputList()
|
| D | xla_op_kernel.h | 113 Status ConstantInput(int index, xla::Literal* constant_literal);
114 Status ConstantInput(absl::string_view name, xla::Literal* constant_literal);
|