| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_outfeed_ops.cc | 54 std::vector<DataType> dtypes; in __anon8aba48030102() local
88 std::vector<DataType> dtypes; in __anon8aba48030202() local
|
| D | resource_variable_ops.cc | 111 DataTypeVector dtypes; in __anon296a55ec0302() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | snapshot_utils_test.cc | 31 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
|
| D | snapshot_utils.cc | 100 const DataTypeVector& dtypes, in Create()
178 const DataTypeVector& dtypes) in CustomWriter()
339 const DataTypeVector& dtypes, in Create()
375 const DataTypeVector& dtypes, in Dataset()
668 const DataTypeVector& dtypes, in MakeNestedDataset()
711 const DataTypeVector& dtypes) in TFRecordReader()
747 const DataTypeVector& dtypes) in CustomReader()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 48 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 …]
|
| D | while_v2.py | 989 dtypes=None, # pylint: disable=redefined-outer-name argument
1046 dtypes=None, # pylint: disable=redefined-outer-name argument
1092 dtypes=None, # pylint: disable=redefined-outer-name argument
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | op_cache.h | 69 llvm::SmallVector<string_view, 4> dtypes) const { in Contains()
81 string_view device_name, llvm::SmallVector<string_view, 4> dtypes) in CacheKey()
|
| D | op_cache.cc | 27 llvm::SmallVector<string_view, 4> dtypes, in GetOrAddOp()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | banded_triangular_solve_op_test.py | 31 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
57 dtypes=(np.float32, np.float64)): argument
|
| D | matrix_triangular_solve_op_test.py | 31 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
57 dtypes=(np.float32, np.float64)): argument
|
| D | denormal_test.py | 37 def _flushDenormalsTest(self, dtypes): argument
|
| D | matrix_band_part_op_test.py | 144 dtypes = (np.bool_, np.int32, np.int64, np.float16, variable
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | variable_merger_pass.cc | 56 DataTypeVector dtypes(num_var_handles); in MergeVarHandleOps() local
101 DataTypeVector dtypes(num_reads); in MergeReadVariableOps() local
|
| D | encapsulate_tpu_computations_pass.cc | 1292 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/ |
| D | cluster_function_library_runtime.cc | 65 DataTypeVector dtypes; in ConstructFunctionGraph() local
117 DataTypeVector dtypes; in ConstructFunctionGraph() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | cluster_scoping_pass_test.cc | 67 std::initializer_list<DataType> dtypes, in BuildStageNode()
|
| /external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/ |
| D | vktSpvAsmComputeShaderCase.cpp | 192 …ayout (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorSetLayout()
242 …rPool (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorPool()
258 …ce, VkDescriptorPool pool, VkDescriptorSetLayout layout, const vector<VkDescriptorType>& dtypes, c… in createDescriptorSet()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_values.cc | 255 const NumpyScalarTypes& dtypes = GetNumpyScalarTypes(); in DevicePut() local
373 const NumpyScalarTypes& dtypes = GetNumpyScalarTypes(); in PyArgSignatureOfValue() local
|
| D | types.cc | 95 NumpyScalarTypes* dtypes = new NumpyScalarTypes(); in GetNumpyScalarTypes() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.cc | 390 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/kernels/ |
| D | partitioned_function_ops.cc | 151 DataTypeVector dtypes; in FillOutputDevices() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | lmdb_dataset_op.cc | 45 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING, DT_STRING}); in output_dtypes() local
|
| D | random_dataset_op.cc | 66 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64}); in output_dtypes() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 100 AttrValue dtypes; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 71 bool* is_type_list, DataTypeVector* dtypes) { in ArgNumType()
178 DataTypeVector dtypes; in BuildInputArgIndex() local
227 DataTypeVector dtypes; in BuildNodeOutputIndex() local
258 DataTypeVector dtypes; in InstantiateNode() local
348 DataTypeVector dtypes; in AddReturnNode() local
414 DataTypeVector dtypes; member
|