Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc139 Status XlaOpKernelContext::ConstantInput(int index, in ConstantInput() function in tensorflow::XlaOpKernelContext
178 Status XlaOpKernelContext::ConstantInput(absl::string_view name, in ConstantInput() function in tensorflow::XlaOpKernelContext
182 return ConstantInput(index, constant_literal, mode); in ConstantInput()
237 TF_RETURN_IF_ERROR(ConstantInput(index, &literal, mode)); in ConstantInputAsIntScalar()
250 TF_RETURN_IF_ERROR(ConstantInput(index, &literal, mode)); in ConstantInputAsFloatScalar()
379 TF_RETURN_IF_ERROR(ConstantInput(index, &literal, mode)); in ConstantInputAsIntVector()
411 TF_RETURN_IF_ERROR(ConstantInput(index, &literal, mode)); in ConstantInputAsInt64Literal()
444 TF_RETURN_IF_ERROR(ConstantInput(index, &literal, mode)); in ConstantInputAsShape()
464 TF_RETURN_IF_ERROR(ConstantInput(index, &literal)); in ConstantInputAsPartialShape()
504 TF_RETURN_IF_ERROR(ConstantInput(i, &(*outputs)[i], mode)); in ConstantInputList()
Dxla_op_kernel.h150 Status ConstantInput(
153 Status ConstantInput(
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc246 bool begin_is_constant = ctx->ConstantInput(1, &begin_literal).ok(); in Compile()
247 bool end_is_constant = ctx->ConstantInput(2, &end_literal).ok(); in Compile()
249 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
527 bool begin_is_constant = ctx->ConstantInput(1, &begin_literal).ok(); in Compile()
528 bool end_is_constant = ctx->ConstantInput(2, &end_literal).ok(); in Compile()
529 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
662 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
663 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
664 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
Dsequence_ops.cc91 OP_REQUIRES_OK(ctx, ctx->ConstantInput( in Compile()
93 OP_REQUIRES_OK(ctx, ctx->ConstantInput( in Compile()
95 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &delta)); in Compile()
Droll_op.cc50 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &axis)); in Compile()
Dreverse_op.cc54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &lax)); in Compile()
Dstateful_random_ops.cc184 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in CompileImpl()
443 OP_REQUIRES_OK(ctx, ctx->ConstantInput(alg_input_idx, &alg_literal)); in Compile()
Dcase_op.cc54 ctx->ConstantInput(0, &branch_index_literal).ok(); in GetPrunedBranchesAndIndex()
Dstateless_random_ops_v2.cc181 TF_RETURN_IF_ERROR(ctx->ConstantInput(alg_input_idx, &alg_literal)); in AlgorithmFromInput()