Home
last modified time | relevance | path

Searched refs:DtypeAndPartialTensorShape (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dresource_handle.h80 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes() const { in dtypes_and_shapes()
84 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes) { in set_dtypes_and_shapes()
138 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes = {},
147 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes = {},
182 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes_;
Dresource_handle.cc35 const std::vector<DtypeAndPartialTensorShape>& dtype_and_shapes) { in DtypeAndShapesToString()
38 for (const DtypeAndPartialTensorShape& dtype_and_shape : dtype_and_shapes) { in DtypeAndShapesToString()
87 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes; in FromProto()
96 dtypes_and_shapes.push_back(DtypeAndPartialTensorShape{dtype, shape}); in FromProto()
132 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes, in MakeRefCountingHandle()
Dresource_mgr.h333 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes = {},
340 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes = {},
353 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes = {},
Dresource_mgr.cc36 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes, in MakeResourceHandle()
Dtensor_shape.h768 struct DtypeAndPartialTensorShape { struct
Dfunction.h702 std::unordered_map<int, DtypeAndPartialTensorShape>
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_mgr.cc91 std::vector<DtypeAndPartialTensorShape>* handle) { in GetMirroredResourceShape()
169 std::vector<DtypeAndPartialTensorShape> resource_dtypes_and_shapes; in SerializeRemoteTensorHandle()
199 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes; in DeserializeRemoteTensorHandle()
205 dtypes_and_shapes.push_back(DtypeAndPartialTensorShape{ in DeserializeRemoteTensorHandle()
Dremote_mgr.h100 std::vector<DtypeAndPartialTensorShape>* handle);
109 RemoteTensorHandleInternal, std::vector<DtypeAndPartialTensorShape>,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.h256 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes);
261 std::vector<DtypeAndPartialTensorShape>* result);
342 std::vector<DtypeAndPartialTensorShape> handle_dtypes_and_shapes_;
Dkernel_and_device.h274 std::unordered_map<int, DtypeAndPartialTensorShape> in KernelAndDeviceFunc()
388 std::unordered_map<int, DtypeAndPartialTensorShape>
Dexecute.cc908 const std::unordered_map<int, DtypeAndPartialTensorShape>& in GetKernelCacheKey()
938 const DtypeAndPartialTensorShape& dtype_and_shape = in GetKernelCacheKey()
961 std::unordered_map<int, DtypeAndPartialTensorShape>& in ExtractFunctionInputInfo()
997 std::vector<DtypeAndPartialTensorShape> resource_dtypes_and_shapes; in ExtractFunctionInputInfo()
1001 const DtypeAndPartialTensorShape& dtype_and_shape = in ExtractFunctionInputInfo()
1078 std::unordered_map<int, DtypeAndPartialTensorShape> in GetOrCreateKernelAndDevice()
1524 {DtypeAndPartialTensorShape{dtype->type(), shape->shape()}}); in StoreResourceDtypesAndShapes()
Dtensor_handle_test.cc142 DtypeAndPartialTensorShape dtype_and_shape = {DT_FLOAT, {2, 2}}; in TEST_F()
185 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes; in TEST_F()
Dtensor_handle.cc171 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes) { in SetResourceHandleDtypeAndShape()
176 std::vector<DtypeAndPartialTensorShape>* result) { in GetResourceHandleDtypesAndShapes()
311 std::vector<DtypeAndPartialTensorShape> dtypes_and_shapes; in CreatePackedHandle()
/external/tensorflow/tensorflow/core/kernels/
Dresource_variable_ops.h39 DtypeAndPartialTensorShape dtype_and_shape_;
Dresource_variable_ops.cc246 std::vector<DtypeAndPartialTensorShape>{dtype_and_shape_}); in VarHandleOp()
256 std::vector<DtypeAndPartialTensorShape>{dtype_and_shape_}, in Compute()
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.cc585 std::unordered_map<int, DtypeAndPartialTensorShape>& in Instantiate()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc892 const DtypeAndPartialTensorShape dtype_and_shape = in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc391 Status SetArgShape(const std::unordered_map<int, DtypeAndPartialTensorShape>& in SetArgShape()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2352 DtypeAndPartialTensorShape dtype_and_shape; in TestConvertReadVariableOp()
2359 std::vector<DtypeAndPartialTensorShape>{dtype_and_shape}); in TestConvertReadVariableOp()