| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | bias_op_deterministic_test.py | 43 data_layout): argument
45 if data_layout == 'channels_first':
47 elif data_layout == 'channels_last':
53 def _dataFormatFromDataLayout(self, data_layout=None): argument
54 if data_layout == 'channels_first':
56 elif data_layout == 'channels_last':
74 data_layout=['channels_first', 'channels_last'],
79 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
86 hash(data_layout) % 256 + hash(data_rank) % 256 +
93 data_dim, data_layout)
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_bytesizeof_test.cc | 23 llvm::DataLayout data_layout( in TEST_F() local
27 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
28 data_layout.getPointerSize(0 /* default address space */)); in TEST_F()
32 llvm::DataLayout data_layout("e-m:e-i64:64-i128:128-n32:64-S128"); in TEST_F() local
35 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
36 data_layout.getPointerSize(0 /* default address space */)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 80 def.data_layout == DataLayout::BHWC; in IsBHWCOpenCLBuffer()
85 def.data_layout == DataLayout::DHWC4; in IsOpenCLTensor()
88 def.data_layout == DataLayout::HDWC4; in IsOpenCLTensor()
91 def.data_layout == DataLayout::DHWC4; in IsOpenCLTensor()
94 def.data_layout == DataLayout::BHWC; in IsOpenCLTensor()
125 input_def.object_def.object_type, input_def.object_def.data_layout); in Init()
134 output_def.object_def.object_type, output_def.object_def.data_layout); in Init()
212 input_def.object_def.object_type, input_def.object_def.data_layout); in Init()
310 output_def.object_def.object_type, output_def.object_def.data_layout); in Init()
380 def.object_def.data_layout)) { in CalculateTextureRegion()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter.cc | 100 output.data_layout == DataLayout::BHWC && in IsSupported()
103 input.data_layout == DataLayout::DHWC4; in IsSupported()
188 input.data_layout == DataLayout::BHWC && in IsSupported()
191 output.data_layout == DataLayout::DHWC4; in IsSupported()
279 input.data_layout == output.data_layout; in IsSupported()
309 input.data_layout == output.data_layout && in IsSupported()
|
| D | converter_test.cc | 73 input_def.object_def.data_layout = DataLayout::DHWC4; in RunFromTensorTest()
77 output_def.object_def.data_layout = DataLayout::BHWC; in RunFromTensorTest()
130 input_def.object_def.data_layout = DataLayout::BHWC; in RunToTensorTest()
134 output_def.object_def.data_layout = DataLayout::DHWC4; in RunToTensorTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | gpu_api_delegate.cc | 62 DataLayout ToDataLayoutFromTFL(TfLiteGpuDataLayout data_layout) { in ToDataLayoutFromTFL() argument
63 switch (data_layout) { in ToDataLayoutFromTFL()
234 DataType data_type, DataLayout data_layout) { in BindGlBufferToTensor() argument
242 def.object_def.data_layout = data_layout; in BindGlBufferToTensor()
258 default_object_def.data_layout = DataLayout::BHWC; in GetObjectDef()
383 TfLiteType data_type, TfLiteGpuDataLayout data_layout) { in TfLiteGpuDelegateBindGlBufferToTensor() argument
395 auto layout = tflite::gpu::cl::ToDataLayoutFromTFL(data_layout); in TfLiteGpuDelegateBindGlBufferToTensor()
|
| D | tensor_type_util.cc | 57 DataLayout data_layout) { in ToTensorStorageType() argument
62 switch (data_layout) { in ToTensorStorageType()
|
| D | gl_interop.h | 148 input.data_layout == output.data_layout && in IsSupported()
|
| D | tensor_type_util.h | 31 DataLayout data_layout);
|
| D | gpu_api_delegate.h | 93 TfLiteType data_type, TfLiteGpuDataLayout data_layout);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.cc | 266 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
270 se::dnn::DataLayout data_layout; in Compute() local
273 data_layout = se::dnn::DataLayout::kBatchYXDepth; in Compute()
276 data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute()
281 data_layout = se::dnn::DataLayout::kBatchYXDepth; in Compute()
367 .set_layout(data_layout); in Compute()
374 .set_layout(data_layout); in Compute()
535 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
555 se::dnn::DataLayout data_layout; in Compute() local
558 data_layout = se::dnn::DataLayout::kBatchYXDepth; in Compute()
[all …]
|
| /external/llvm-project/mlir/test/Conversion/StandardToLLVM/ |
| D | convert-data-layout.mlir | 4 // CHECK: module attributes {llvm.data_layout = ""}
5 // CHECK-32: module attributes {llvm.data_layout ="p:32:32:32"}
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_init.c | 295 char *data_layout; in init_gallivm_engine() local
298 data_layout = LLVMCopyStringRepOfTargetData(gallivm->target); in init_gallivm_engine()
302 debug_printf("module target data = %s\n", data_layout); in init_gallivm_engine()
306 free(data_layout); in init_gallivm_engine()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | api.h | 175 DataLayout data_layout = DataLayout::UNKNOWN; member
186 return data_type == other.data_type && data_layout == other.data_layout &&
|
| D | api.cc | 65 def.data_layout != DataLayout::UNKNOWN && in IsValid()
107 switch (def.object_def.data_layout) { in NumElements()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 46 const char* data_layout);
155 const std::string& target_triple, const std::string& data_layout,
|
| D | gpu_compiler.cc | 141 const char* target_triple, const char* data_layout) in GpuCompiler() argument
144 data_layout_(data_layout), in GpuCompiler()
145 pointer_size_(llvm::DataLayout(data_layout) in GpuCompiler()
550 const std::string& target_triple, const std::string& data_layout, in CompileModuleToLlvmIrImpl() argument
562 (*llvm_module)->setDataLayout(data_layout); in CompileModuleToLlvmIrImpl()
985 const std::string& target_triple, const std::string& data_layout, in CompileModuleToLlvmIr() argument
994 hlo_module, llvm_context, target_triple, data_layout, platform_name, in CompileModuleToLlvmIr()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api2.cc | 168 external_def.data_layout == internal_def.data_layout) || in IsSameDef()
172 external_def.data_layout == DataLayout::BHWC && in IsSameDef()
173 internal_def.data_layout == DataLayout::DHWC4 && in IsSameDef()
341 inner_def.internal_def.object_def.data_layout = DataLayout::DHWC4; in MakeOuterInnerDefs()
622 external_def.object_def.data_layout = DataLayout::DHWC4; in LinkTensors()
|
| /external/llvm-project/llvm/bindings/ocaml/executionengine/ |
| D | llvm_executionengine.mli | 69 (** [data_layout ee] is the data layout of the execution engine [ee]. *)
70 val data_layout : llexecutionengine -> Llvm_target.DataLayout.t val
|
| D | llvm_executionengine.ml | 43 external data_layout : llexecutionengine -> Llvm_target.DataLayout.t
|
| /external/llvm/bindings/ocaml/executionengine/ |
| D | llvm_executionengine.mli | 70 (** [data_layout ee] is the data layout of the execution engine [ee]. *)
71 val data_layout : llexecutionengine -> Llvm_target.DataLayout.t val
|
| D | llvm_executionengine.ml | 44 external data_layout : llexecutionengine -> Llvm_target.DataLayout.t
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | simple_orc_jit.h | 76 const llvm::DataLayout& data_layout() const { return data_layout_; } in data_layout() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | internal_api_samples.cc | 176 obj_def.data_layout = DataLayout::BHWC; in RunModelSampleWithInternalAPI()
305 obj_def.data_layout = DataLayout::BHWC; in RunModelSampleWithInternalAPISerializedKernels()
381 obj_def.data_layout = DataLayout::BHWC; in RunModelSampleWithInternalAPISerialized()
|
| /external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/ |
| D | IRDynamicChecks.cpp | 286 llvm::DataLayout data_layout(&m_module); in GetIntptrTy() local
289 data_layout.getPointerSizeInBits()); in GetIntptrTy()
|