Searched refs:output_tensor_info (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | signature_def_utils_impl.py | 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}
|
| D | save_test.py | 72 for output_name, output_tensor_info in signature.outputs.items():
74 output_tensor_info.name)
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 42 std::vector<TapeTensor> output_tensor_info; member
460 for (const auto& output : op_it->second.output_tensor_info) { in DeleteTrace()
616 for (int j = 0; j < op_it->second.output_tensor_info.size(); ++j) { in InitialGradients()
617 if (op_it->second.output_tensor_info[j].GetID() == id) { in InitialGradients()
621 op_it->second.output_tensor_info[j], &ones_like)); in InitialGradients()
733 out_gradients.reserve(trace.output_tensor_info.size()); in ComputeGradient()
745 for (int i = 0, end = trace.output_tensor_info.size(); i < end; ++i) { in ComputeGradient()
746 const int64 id = trace.output_tensor_info[i].GetID(); in ComputeGradient()
781 out_gradients[i] = trace.output_tensor_info[i].ZerosLike(); in ComputeGradient()
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | saved_model_test.py | 79 for output_name, output_tensor_info in signature.outputs.items():
81 output_tensor_info.name)
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils_test.py | 266 for output_name, output_tensor_info in signature.outputs.items():
268 output_tensor_info.name)
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer_test.cc | 303 RemoteFusedGraphExecuteUtils::TensorShapeMap output_tensor_info; in TEST_F() local
305 def, inputs, {}, &output_tensor_info); in TEST_F()
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 522 for output_name, output_tensor_info in signature.outputs.items():
524 output_tensor_info.name)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 277 def output_tensor_info(self) -> Mapping[str, meta_graph_pb2.TensorInfo]: member in ModelHandlerV1
287 return [info.name for info in self.output_tensor_info.values()]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 63 for output_name, output_tensor_info in signature.outputs.items():
65 output_tensor_info.name)
|