Searched refs:output_tensor_shapes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 415 std::vector<TensorShape> output_tensor_shapes; in AllocateOutputTensors() local
416 output_tensor_shapes.reserve(sub_elements); in AllocateOutputTensors()
429 output_tensor_shapes.push_back(shape); in AllocateOutputTensors()
452 output_tensor_shapes[i] = shape; in AllocateOutputTensors()
486 TF_RET_CHECK(num_updated_variables <= output_tensor_shapes.size()) in AllocateOutputTensors()
487 << num_updated_variables << " <= " << output_tensor_shapes.size(); in AllocateOutputTensors()
521 for (int i = 0; i < output_tensor_shapes.size(); ++i) { in AllocateOutputTensors()
530 op_output_index, output_tensor_shapes[i], &output_tensor)); in AllocateOutputTensors()
555 output_tensor_shapes[i], in AllocateOutputTensors()
569 op_output_index, output_tensor_shapes[i], &output_tensor)); in AllocateOutputTensors()
[all …]
|
| D | tpu_reshard_variables_op_util.cc | 246 std::vector<TensorShape> output_tensor_shapes; in UpdateOutputVariables() local
247 output_tensor_shapes.reserve(sub_elements); in UpdateOutputVariables()
260 output_tensor_shapes.push_back(shape); in UpdateOutputVariables()
301 variables[i].var()->tensor()->dtype(), output_tensor_shapes[i], in UpdateOutputVariables()
|
| D | tpu_executable_info.proto | 53 repeated TensorShapeProto output_tensor_shapes = 8; field
|
| D | tpu_reshard_variables_op.cc | 270 context, std::move(result), executable->output_tensor_shapes(), backend, in DoTpuExecute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 501 std::vector<TensorShape> output_tensor_shapes; in PopulateOutputs() local
502 output_tensor_shapes.reserve(ctx->num_outputs()); in PopulateOutputs()
526 output_tensor_shapes.push_back(shape); in PopulateOutputs()
530 output_tensor_shapes.push_back(compilation_result->outputs[i].shape); in PopulateOutputs()
537 const TensorShape& shape = output_tensor_shapes[i]; in PopulateOutputs()
|