| /external/tensorflow/tensorflow/python/framework/ |
| D | fast_tensor_util.pyx | 10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
53 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray): argument
60 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray): argument
67 tensor_proto, np.ndarray[np.uint64_t, ndim=1] nparray): argument
74 tensor_proto, np.ndarray[np.uint8_t, ndim=1] nparray): argument
82 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
[all …]
|
| D | tensor_util.py | 50 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
55 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
67 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
72 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
126 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
129 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values): argument
132 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values): argument
135 def SlowAppendInt64ArrayToTensorProto(tensor_proto, proto_values): argument
138 def SlowAppendQIntArrayToTensorProto(tensor_proto, proto_values): argument
141 def SlowAppendUInt32ArrayToTensorProto(tensor_proto, proto_values): argument
[all …]
|
| D | op_def_util.cc | 170 static PyObject* tensor_proto = GetRegisteredPyObject("tf.TensorProto"); in operator ()() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op_test.cc | 41 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
54 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
115 TensorProto tensor_proto; in GetRandomStringsTensorProto() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util_test.cc | 438 TensorProto tensor_proto = in TEST() local
460 TensorProto tensor_proto = in TEST() local
526 TensorProto tensor_proto; in CreateAsProtoTensorContent() local
537 TensorProto tensor_proto; in CreateAsProtoField() local
557 TensorProto tensor_proto = in ConstantTailTest() local
|
| D | device_base.h | 238 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 61 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| D | threadpool_device.cc | 89 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | renamed_device.h | 106 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| D | constant_folding_test.cc | 54 const TensorProto* tensor_proto; in ExpectNodeClose() local
67 const TensorProto* tensor_proto; in ExpectNodeEqual() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 50 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 475 const TensorProto& tensor_proto, in MakeTensorFromProto()
499 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
513 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeFastMemTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 63 TensorProto tensor_proto; in ConvertToProto() local
376 TensorProto tensor_proto; in ConvertToTensor() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_grpc_testlib.cc | 54 const TensorProto& tensor_proto = val.tensor(); in SendEvents() local
|
| D | debug_io_utils.cc | 129 TensorProto* tensor_proto, in WrapStringTensorAsEvents()
190 TensorProto tensor_proto; in WrapTensorAsEvents() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 147 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.h | 86 TensorProto tensor_proto; in SetNodeTensorAttr() local
|
| D | transform_utils_test.cc | 186 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttr() local
202 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttrWithTensor() local
212 TensorProto tensor_proto; in TestGetNodeTensorAttr() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 398 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
418 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto()
1004 const auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
1088 const TensorProto* tensor_proto = ctx->input_tensor_protos[dst_input]; in UpdateNode() local
1586 TensorProto tensor_proto; in UpdateOutputShapesAndValues() local
1676 const TensorProto& tensor_proto = node.attr().at("value").tensor(); in MaybeUpdateNodeContextOutput() local
1928 TensorProto tensor_proto; in MakeIntegerScalarTensorProto() local
1941 const TensorProto& tensor_proto, in MaybeTensorProtoToShape()
|
| D | utils.cc | 67 for (const auto& tensor_proto : attr_value.list().tensor()) { in ExtractTensors() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 203 TensorProto tensor_proto = TensorFromShape(shape_proto); in ReplaceBroadcast() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | debug_event.proto | 296 TensorProto tensor_proto = 5; field
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.cc | 241 for (const auto& tensor_proto : response->tensor()) { in Run() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 79 const TensorProto& tensor_proto = attr_name_and_value->second.tensor(); in ConstantFromSavedConstant() local
|