Searched refs:output_tensor_info (Results 1 – 6 of 6) sorted by relevance
36 std::vector<TapeTensor> output_tensor_info; member260 for (const auto& output : op_it->second.output_tensor_info) { in DeleteTrace()416 for (int j = 0; j < op_it->second.output_tensor_info.size(); ++j) { in InitialGradients()417 if (op_it->second.output_tensor_info[j].GetID() == id) { in InitialGradients()420 vspace.Ones(op_it->second.output_tensor_info[j])); in InitialGradients()526 out_gradients.reserve(trace.output_tensor_info.size()); in ComputeGradient()528 for (int i = 0; i < trace.output_tensor_info.size(); ++i) { in ComputeGradient()529 const int64 id = trace.output_tensor_info[i].GetID(); in ComputeGradient()538 out_gradients.push_back(vspace.Zeros(trace.output_tensor_info[i])); in ComputeGradient()
100 output_tensor_info = utils.build_tensor_info(predictions)101 if output_tensor_info.dtype != types_pb2.DT_FLOAT:103 signature_outputs = {signature_constants.REGRESS_OUTPUTS: output_tensor_info}
81 for output_name, output_tensor_info in signature.outputs.items():83 output_tensor_info.name)
94 TensorInfo output_tensor_info = in LoadAndValidateSavedModelBundle() local97 EXPECT_EQ(DT_FLOAT, output_tensor_info.dtype()); in LoadAndValidateSavedModelBundle()99 output_tensor_info.name()); in LoadAndValidateSavedModelBundle()
185 for output_name, output_tensor_info in signature.outputs.items():187 output_tensor_info.name)
303 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info; in TEST_F() local305 def, inputs, {}, &output_tensor_info); in TEST_F()