| /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 | 48 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
53 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
65 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
70 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
124 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
127 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values): argument
130 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values): argument
133 def SlowAppendInt64ArrayToTensorProto(tensor_proto, proto_values): argument
136 def SlowAppendQIntArrayToTensorProto(tensor_proto, proto_values): argument
139 def SlowAppendUInt32ArrayToTensorProto(tensor_proto, proto_values): argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op_test.cc | 40 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
53 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
107 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 | 248 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.cc | 45 Status SYCLDevice::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 | 93 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| D | renamed_device.h | 116 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()
506 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
520 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeFastMemTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 59 TensorProto tensor_proto; in ConvertToProto() local
330 TensorProto tensor_proto; in ConvertToTensor() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 129 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, 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/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 | 373 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
393 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto()
810 const auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
925 const TensorProto* tensor_proto = ctx->input_tensor_protos[dst_input]; in UpdateNode() local
1408 TensorProto tensor_proto; in UpdateOutputShapesAndValues() local
1491 const TensorProto& tensor_proto = node.attr().at("value").tensor(); in MaybeUpdateNodeContextOutput() local
1738 TensorProto tensor_proto; in MakeIntegerScalarTensorProto() local
1751 const TensorProto& tensor_proto, in MaybeTensorProtoToShape()
|
| D | utils.cc | 67 for (const auto& tensor_proto : attr_value.list().tensor()) { in ExtractTensors() local
|
| D | op_level_cost_estimator_test.cc | 371 TensorProto* tensor_proto) { in GetTensorProto()
419 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp() local
1066 TensorProto tensor_proto; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | debug_event.proto | 291 TensorProto tensor_proto = 5; field
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 203 TensorProto tensor_proto = TensorFromShape(shape_proto); in ReplaceBroadcast() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 523 for (const auto& tensor_proto : send_tensor.tensors()) { in SendTensor() local
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 737 const TensorProto* tensor_proto = nullptr; in RegisterPadNode() local
1101 const TensorProto& tensor_proto, Tensor* tensor) { in MakeTensorFromProto()
|