Searched refs:SetConstantOutput (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 215 ctx->SetConstantOutput(5, Tensor()); in Compile()
229 ctx->SetConstantOutput(5, Tensor()); in Compile()
339 ctx->SetConstantOutput(3, Tensor()); in Compile()
340 ctx->SetConstantOutput(4, Tensor()); in Compile()
|
| D | bcast_ops.cc | 63 ctx->SetConstantOutput(0, output); in Compile()
118 ctx->SetConstantOutput(idx, constant); in Output()
|
| D | shape_op.cc | 61 ctx->SetConstantOutput(0, shape_constant); in Compile()
173 ctx->SetConstantOutput(i, shape_constant); in Compile()
195 ctx->SetConstantOutput(0, rank_constant); in Compile()
|
| D | tensor_array_ops.cc | 177 ctx->SetConstantOutput(1, flow); in Compile()
471 ctx->SetConstantOutput(1, lengths); in Compile()
562 ctx->SetConstantOutput(0, size_tensor); in Compile()
595 ctx->SetConstantOutput(1, Tensor(DT_FLOAT)); in Compile()
|
| D | const_op.cc | 123 ctx->SetConstantOutput(0, tensor); in Compile()
|
| D | concat_op.cc | 198 ctx->SetConstantOutput(i, out_constant); in Compile()
|
| D | variable_ops.cc | 56 ctx->SetConstantOutput(0, shape_constant); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 190 void SetConstantOutput(int index, const Tensor& host_tensor);
|
| D | graph_compiler.cc | 303 xla_op_context.SetConstantOutput(i, result.outputs[i].constant_value); in CompileFunctionalNode()
|
| D | xla_op_kernel.cc | 594 void XlaOpKernelContext::SetConstantOutput(int index, const Tensor& constant) { in SetConstantOutput() function in tensorflow::XlaOpKernelContext
|