Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/runtime/
Dsymbolic_shape_test.cc39 llvm::SmallVector<DType> dtypes, in GetFunctionType()
83 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST() local
146 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST() local
208 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST() local
282 auto dtypes = {DType::F32, DType::I32}; in TEST() local
306 auto dtypes = {DType::F32, DType::I32}; in TEST() local
345 auto dtypes = {DType::F32, DType::I32}; in TEST() local
371 auto dtypes = {DType::F32}; in TEST() local
468 auto dtypes = {DType::F32, DType::I32, DType::I1, DType::F32}; in BenchmarkFullyDynamic() local
492 auto dtypes = {DType::F32, DType::I32, DType::I1, DType::F32}; in BenchmarkSameDynamic() local
[all …]
/external/tensorflow/tensorflow/core/ops/
Dtpu_outfeed_ops.cc54 std::vector<DataType> dtypes; in __anona6b0b89c0102() local
88 std::vector<DataType> dtypes; in __anona6b0b89c0202() local
Dresource_variable_ops.cc126 DataTypeVector dtypes; in __anon620315450302() local
/external/tensorflow/tensorflow/core/data/
Dsnapshot_utils_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_utils.cc100 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/
Ddata_flow_ops.py44 def _as_type_list(dtypes): argument
56 dtypes, argument
88 def _as_name_list(names, dtypes): argument
136 def __init__(self, dtypes, shapes, names, queue_ref): argument
240 def dtypes(self): member in QueueBase
637 dtypes, argument
722 dtypes, argument
786 dtypes, argument
863 dtypes, argument
1613 dtypes, argument
[all …]
Dwhile_v2.py1001 dtypes=None, # pylint: disable=redefined-outer-name argument
1058 dtypes=None, # pylint: disable=redefined-outer-name argument
1104 dtypes=None, # pylint: disable=redefined-outer-name argument
/external/tensorflow/tensorflow/core/tfrt/eager/
Dop_cache.h69 llvm::SmallVector<string_view, 4> dtypes) const { in Contains()
81 string_view device_name, llvm::SmallVector<string_view, 4> dtypes) in CacheKey()
Dop_cache.cc27 llvm::SmallVector<string_view, 4> dtypes, in GetOrAddOp()
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dbanded_triangular_solve_op_test.py27 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
53 dtypes=(np.float32, np.float64)): argument
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dmatrix_triangular_solve_op_test.py27 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
53 dtypes=(np.float32, np.float64)): argument
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc56 DataTypeVector dtypes(num_var_handles); in MergeVarHandleOps() local
99 DataTypeVector dtypes(num_reads); in MergeReadVariableOps() local
Dencapsulate_tpu_computations_pass.cc1255 std::vector<DataType> dtypes(num_replicas, dtype); in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1661 std::vector<DataType> dtypes; in FindArgsToLiftForIfNode() local
1724 std::vector<DataType> dtypes; in FindArgsToLiftForWhileNode() local
1807 std::vector<DataType> dtypes(call_node->input_types().begin(), in FindArgsToLiftForCallNode() local
1980 const std::vector<DataType>& dtypes, Node* n) { in UpdateTypeAttribute()
2062 const std::vector<DataType>& dtypes, FunctionBody* fbody) { in CleanUpRetvalsForWhileBody()
2091 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromWhileNode() local
2158 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromIfNode() local
2249 std::vector<DataType> dtypes; in LiftOutsideCompilationOnlyArgsFromCallNode() local
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Ddenormal_test.py36 def _flushDenormalsTest(self, dtypes): argument
Dmatrix_band_part_op_test.py140 dtypes = (np.bool_, np.int32, np.int64, np.float16, variable
/external/tensorflow/tensorflow/c/experimental/ops/
Dio_ops.cc54 absl::Span<DataType> dtypes, const char* name, in RestoreV2()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc65 DataTypeVector dtypes; in ConstructFunctionGraph() local
116 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.cpp192 …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/
Dpy_values.cc252 const NumpyScalarTypes& dtypes = GetNumpyScalarTypes(); in DevicePut() local
370 const NumpyScalarTypes& dtypes = GetNumpyScalarTypes(); in PyArgSignatureOfValue() local
Dtypes.cc96 NumpyScalarTypes* dtypes = new NumpyScalarTypes(); in GetNumpyScalarTypes() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc392 const OpDef::ArgDef& input_def, const std::vector<DataType>& dtypes) { in InferMixedTypeInputListAttrs()
408 std::vector<DataType> dtypes(num_inputs); in InferInputListAttrs() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlmdb_dataset_op.cc45 static DataTypeVector* dtypes = new DataTypeVector({DT_STRING, DT_STRING}); in output_dtypes() local
Drandom_dataset_op.cc65 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64}); in output_dtypes() local
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc152 DataTypeVector dtypes; in FillOutputDevices() local

123