Home
last modified time | relevance | path

Searched refs:constant_literal (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h113 Status ConstantInput(int index, xla::Literal* constant_literal);
114 Status ConstantInput(absl::string_view name, xla::Literal* constant_literal);
275 xla::Literal* constant_literal);
Dxla_op_kernel.cc114 xla::Literal* constant_literal) { in ConstantInput() argument
116 index, context_->input(index).shape().dim_sizes(), constant_literal); in ConstantInput()
133 xla::Literal* constant_literal) { in ConstantInput() argument
135 return ConstantInput(index, constant_literal); in ConstantInput()
140 xla::Literal* constant_literal) { in ConstantInputReshaped() argument
174 TF_ASSIGN_OR_RETURN(*constant_literal, HostTensorToLiteral(temp)); in ConstantInputReshaped()