Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/ops/
Dtpu_outfeed_ops.cc54 std::vector<DataType> dtypes; in __anon0c58097e0102() local
88 std::vector<DataType> dtypes; in __anon0c58097e0202() local
Dresource_variable_ops.cc111 DataTypeVector dtypes; in __anon50d007a70302() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util_test.cc31 void GenerateTensorVector(tensorflow::DataTypeVector& dtypes, in GenerateTensorVector()
44 tensorflow::DataTypeVector dtypes; in SnapshotRoundTrip() local
96 tensorflow::DataTypeVector dtypes; in SnapshotReaderBenchmarkLoop() local
152 tensorflow::DataTypeVector dtypes; in SnapshotWriterBenchmarkLoop() local
Dsnapshot_util.cc84 const DataTypeVector& dtypes, in Create()
162 const DataTypeVector& dtypes) in CustomWriter()
323 const DataTypeVector& dtypes, in Create()
358 const int64 version, const DataTypeVector& dtypes, in Dataset()
578 const DataTypeVector& dtypes, in MakeNestedDataset()
617 const DataTypeVector& dtypes) in TFRecordReader()
653 const DataTypeVector& dtypes) in CustomReader()
Drandom_dataset_op.cc50 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64}); in output_dtypes() local
Dlmdb_dataset_op.cc45 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING, DT_STRING}); in output_dtypes() 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
784 dtypes, argument
861 dtypes, argument
1612 dtypes, argument
[all …]
Dwhile_v2.py952 dtypes=None, # pylint: disable=redefined-outer-name argument
1009 dtypes=None, # pylint: disable=redefined-outer-name argument
1055 dtypes=None, # pylint: disable=redefined-outer-name argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dbanded_triangular_solve_op_test.py31 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
57 dtypes=(np.float32, np.float64)): argument
Dmatrix_triangular_solve_op_test.py31 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
57 dtypes=(np.float32, np.float64)): argument
Ddenormal_test.py37 def _flushDenormalsTest(self, dtypes): argument
Dmatrix_band_part_op_test.py144 dtypes = (np.bool, np.int32, np.int64, np.float16, variable
Dsvd_op_test.py388 dtypes = ([np.float32, np.float64] + [np.complex64, np.complex128] * variable
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc56 DataTypeVector dtypes(num_var_handles); in MergeVarHandleOps() local
101 DataTypeVector dtypes(num_reads); in MergeReadVariableOps() local
Dencapsulate_tpu_computations_pass.cc1292 std::vector<DataType> dtypes(num_replicas, dtype); in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1703 std::vector<DataType> dtypes; in FindArgsToLiftForIfNode() local
1766 std::vector<DataType> dtypes; in FindArgsToLiftForWhileNode() local
1849 std::vector<DataType> dtypes(call_node->input_types().begin(), in FindArgsToLiftForCallNode() local
2026 const std::vector<DataType>& dtypes, Node* n) { in UpdateTypeAttribute()
2108 const std::vector<DataType>& dtypes, FunctionBody* fbody) { in CleanUpRetvalsForWhileBody()
2137 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromWhileNode() local
2204 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromIfNode() local
2295 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromCallNode() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc65 DataTypeVector dtypes; in ConstructFunctionGraph() local
117 DataTypeVector dtypes; in ConstructFunctionGraph() local
/external/tensorflow/tensorflow/compiler/jit/
Dcluster_scoping_pass_test.cc67 std::initializer_list<DataType> dtypes, in BuildStageNode()
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmComputeShaderCase.cpp190 …ayout (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorSetLayout()
240 …rPool (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorPool()
256 …ce, VkDescriptorPool pool, VkDescriptorSetLayout layout, const vector<VkDescriptorType>& dtypes, c… in createDescriptorSet()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc151 DataTypeVector dtypes; in FillOutputDevices() local
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_dataset_op.cc90 AttrValue dtypes; in AsGraphDefInternal() local
Dtensor_slice_dataset_op.cc103 AttrValue dtypes; in AsGraphDefInternal() local
Dtext_line_dataset_op.cc59 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING}); in output_dtypes() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc390 const OpDef::ArgDef& input_def, const std::vector<DataType>& dtypes) { in InferMixedTypeInputListAttrs()
406 std::vector<DataType> dtypes(num_inputs); in InferInputListAttrs() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc70 bool* is_type_list, DataTypeVector* dtypes) { in ArgNumType()
176 DataTypeVector dtypes; in BuildInputArgIndex() local
225 DataTypeVector dtypes; in BuildNodeOutputIndex() local
256 DataTypeVector dtypes; in InstantiateNode() local
346 DataTypeVector dtypes; in AddReturnNode() local
412 DataTypeVector dtypes; member
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py221 def infeed_dequeue_tuple(dtypes, shapes, name=None): argument

123