Searched refs:ConstantInput (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 130 Status XlaOpKernelContext::ConstantInput(int index, in ConstantInput() function in tensorflow::XlaOpKernelContext
149 Status XlaOpKernelContext::ConstantInput(absl::string_view name, in ConstantInput() function in tensorflow::XlaOpKernelContext
152 return ConstantInput(index, constant_literal); in ConstantInput()
230 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsIntScalar()
242 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsFloatScalar()
345 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsIntVector()
375 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsInt64Literal()
407 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsShape()
417 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsPartialShape()
456 TF_RETURN_IF_ERROR(ConstantInput(i, &(*outputs)[i])); in ConstantInputList()
|
| D | xla_op_kernel.h | 128 Status ConstantInput(int index, xla::Literal* constant_literal);
129 Status ConstantInput(absl::string_view name, xla::Literal* constant_literal);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 209 bool begin_is_constant = ctx->ConstantInput(1, &begin_literal).ok(); in Compile()
210 bool end_is_constant = ctx->ConstantInput(2, &end_literal).ok(); in Compile()
212 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
488 bool begin_is_constant = ctx->ConstantInput(1, &begin_literal).ok(); in Compile()
489 bool end_is_constant = ctx->ConstantInput(2, &end_literal).ok(); in Compile()
490 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
623 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
624 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
625 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
|
| D | sequence_ops.cc | 87 OP_REQUIRES_OK(ctx, ctx->ConstantInput(0, &start)); in Compile()
88 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &limit)); in Compile()
89 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &delta)); in Compile()
|
| D | roll_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &axis)); in Compile()
|
| D | reverse_op.cc | 54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &lax)); in Compile()
|
| D | stateful_random_ops.cc | 183 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in CompileImpl()
442 OP_REQUIRES_OK(ctx, ctx->ConstantInput(alg_input_idx, &alg_literal)); in Compile()
|
| D | case_op.cc | 49 ctx->ConstantInput(0, &branch_index_literal).ok(); in GetPrunedBranchesAndIndex()
|
| D | stateless_random_ops_v2.cc | 178 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in GetAlgorithm()
|