Home
last modified time | relevance | path

Searched refs:ConstantInput (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc54 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()
Dsequence_ops.cc86 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()
Dreverse_op.cc54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &lax)); in Compile()
Dstateful_random_ops.cc193 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in CompileImpl()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc113 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()
Dxla_op_kernel.h113 Status ConstantInput(int index, xla::Literal* constant_literal);
114 Status ConstantInput(absl::string_view name, xla::Literal* constant_literal);