Searched refs:output_tensor_shapes (Results 1 – 3 of 3) sorted by relevance
415 std::vector<TensorShape> output_tensor_shapes; in AllocateOutputTensors() local416 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()560 var.var()->tensor()->dtype(), output_tensor_shapes[i], &unused, in AllocateOutputTensors()573 op_output_index, output_tensor_shapes[i], &output_tensor)); in AllocateOutputTensors()[all …]
53 repeated TensorShapeProto output_tensor_shapes = 8; field
461 std::vector<TensorShape> output_tensor_shapes; in PopulateOutputs() local462 output_tensor_shapes.reserve(ctx->num_outputs()); in PopulateOutputs()478 output_tensor_shapes.push_back(shape); in PopulateOutputs()482 output_tensor_shapes.push_back(compilation_result->outputs[i].shape); in PopulateOutputs()489 const TensorShape& shape = output_tensor_shapes[i]; in PopulateOutputs()