Searched refs:SetConstantOutput (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 219 ctx->SetConstantOutput(5, Tensor()); in Compile()
233 ctx->SetConstantOutput(5, Tensor()); in Compile()
343 ctx->SetConstantOutput(3, Tensor()); in Compile()
344 ctx->SetConstantOutput(4, Tensor()); in Compile()
|
| D | bcast_ops.cc | 69 ctx->SetConstantOutput(0, output); in Compile()
131 ctx->SetConstantOutput(idx, constant); in Output()
|
| D | tensor_array_ops.cc | 177 ctx->SetConstantOutput(1, flow); in Compile()
472 ctx->SetConstantOutput(1, lengths); in Compile()
563 ctx->SetConstantOutput(0, size_tensor); in Compile()
596 ctx->SetConstantOutput(1, Tensor(DT_FLOAT)); in Compile()
|
| D | shape_op.cc | 61 ctx->SetConstantOutput(0, shape_constant); in Compile()
201 ctx->SetConstantOutput(i, shape_constant); in Compile()
223 ctx->SetConstantOutput(0, rank_constant); in Compile()
|
| D | const_op.cc | 132 ctx->SetConstantOutput(0, tensor); in Compile()
|
| D | concat_op.cc | 202 ctx->SetConstantOutput(i, out_constant); in Compile()
|
| D | variable_ops.cc | 69 ctx->SetConstantOutput(0, shape_constant); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 240 void SetConstantOutput(int index, const Tensor& host_tensor);
|
| D | graph_compiler.cc | 304 xla_op_context.SetConstantOutput(i, result.outputs[i].constant_value); in CompileFunctionalNode()
|
| D | xla_op_kernel.cc | 675 void XlaOpKernelContext::SetConstantOutput(int index, const Tensor& constant) { in SetConstantOutput() function in tensorflow::XlaOpKernelContext
|