Home
last modified time | relevance | path

Searched defs:dtypes (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc41 DataTypeVector dtypes, std::vector<PartialTensorShape> shapes, in CreateTensorSliceDatasetKernel()
140 DataTypeVector dtypes; in TEST_F() local
210 DataTypeVector dtypes({DT_INT64, DT_INT64}); in TEST_F() local
239 DataTypeVector dtypes; in TEST_F() local
282 DataTypeVector dtypes; in TEST_F() local
326 DataTypeVector dtypes; in TEST_F() local
362 DataTypeVector dtypes({DT_INT64, DT_INT64}); in TEST_F() local
397 DataTypeVector dtypes; in TEST_F() local
446 DataTypeVector dtypes; in TEST_F() local
494 DataTypeVector dtypes({DT_INT64, DT_INT64}); in TEST_F() local
[all …]
Ddataset_test_base.cc39 StringPiece node_name, const DataTypeVector& dtypes, in CreateTensorSliceDatasetKernel()
60 DataTypeVector dtypes; in CreateTensorSliceDataset() local
Dtensor_dataset_op.cc89 AttrValue dtypes; in AsGraphDefInternal() local
Drange_dataset_op.cc61 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64}); in output_dtypes() local
Dreader_dataset_ops.cc102 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
367 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
794 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
Dtensor_slice_dataset_op.cc114 AttrValue dtypes; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py48 def _as_type_list(dtypes): argument
60 dtypes, argument
91 def _as_name_list(names, dtypes): argument
138 def __init__(self, dtypes, shapes, names, queue_ref): argument
238 def dtypes(self): member in QueueBase
635 dtypes, argument
720 dtypes, argument
783 dtypes, argument
1496 dtypes, argument
1539 def dtypes(self): member in BaseStagingArea
[all …]
Dmatmul_benchmark_test.py113 dtypes = [np.float32, np.float64] variable
/external/tensorflow/tensorflow/core/ops/
Dtpu_outfeed_ops.cc53 std::vector<DataType> dtypes; in __anon71ff5a600102() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_triangular_solve_op_test.py32 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
58 dtypes=(np.float32, np.float64)): argument
Ddenormal_test.py38 def _flushDenormalsTest(self, use_gpu, dtypes): argument
Dmatrix_band_part_op_test.py144 dtypes = (np.bool, np.int32, np.int64, np.float16, variable
/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
Dignite_dataset_ops.cc26 Status SchemaToTypes(const std::vector<int32>& schema, DataTypeVector* dtypes) { in SchemaToTypes()
179 DataTypeVector dtypes; in MakeDataset() local
Dignite_dataset.cc27 DataTypeVector dtypes, in IgniteDataset()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc38 DataTypeVector dtypes; in ConstructFunctionGraph() local
84 DataTypeVector dtypes; in ConstructFunctionGraph() local
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_prefix_key_dataset_op.cc59 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
Dbigtable_sample_keys_dataset_op.cc52 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
Dbigtable_range_key_dataset_op.cc64 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
Dbigtable_sample_key_pairs_dataset_op.cc77 static DataTypeVector* dtypes = in output_dtypes() local
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py359 dtypes, # pylint: disable=redefined-outer-name argument
387 dtypes=None, # pylint: disable=redefined-outer-name argument
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dparallel_reader.py199 dtypes=None, argument
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmComputeShaderCase.cpp182 …ayout (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorSetLayout()
232 …rPool (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorPool()
248 …ce, VkDescriptorPool pool, VkDescriptorSetLayout layout, const vector<VkDescriptorType>& dtypes, c… in createDescriptorSet()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op.cc64 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64}); in output_dtypes() local
Dlmdb_dataset_op.cc59 static DataTypeVector* dtypes = in output_dtypes() local
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc140 DataTypeVector dtypes; in FillOutputDevices() local

12