Searched refs:SetConstantOutput (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 42 ctx->SetConstantOutput(0, shape_constant); in Compile()
62 ctx->SetConstantOutput(i, shape_constant); in Compile()
83 ctx->SetConstantOutput(0, rank_constant); in Compile()
|
| D | bcast_ops.cc | 63 ctx->SetConstantOutput(0, output); in Compile()
118 ctx->SetConstantOutput(idx, constant); in Output()
|
| D | batch_norm_op.cc | 213 ctx->SetConstantOutput(3, Tensor()); in Compile()
214 ctx->SetConstantOutput(4, Tensor()); 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 | 196 ctx->SetConstantOutput(i, out_constant); in Compile()
|
| D | sequence_ops.cc | 190 ctx->SetConstantOutput(0, out_constant); in Compile()
|
| D | variable_ops.cc | 55 ctx->SetConstantOutput(0, shape_constant); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 175 void SetConstantOutput(int index, const Tensor& host_tensor);
|
| D | graph_compiler.cc | 292 xla_op_context.SetConstantOutput(i, result.outputs[i].constant_value); in CompileFunctionalNode()
|
| D | xla_op_kernel.cc | 487 void XlaOpKernelContext::SetConstantOutput(int index, const Tensor& constant) { in SetConstantOutput() function in tensorflow::XlaOpKernelContext
|