Home
last modified time | relevance | path

Searched defs:data_layout (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_bytesizeof_test.cc23 llvm::DataLayout data_layout( in TEST_F() local
32 llvm::DataLayout data_layout("e-m:e-i64:64-i128:128-n32:64-S128"); in TEST_F() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_deterministic_test.py41 data_layout): argument
51 def _data_format_from_data_layout(self, data_layout=None): argument
81 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc186 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
190 se::dnn::DataLayout data_layout; in Compute() local
373 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
393 se::dnn::DataLayout data_layout; in Compute() local
Dfused_batch_norm_op.cc794 const se::dnn::DataLayout data_layout = in operator ()() local
993 const se::dnn::DataLayout data_layout = in operator ()() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_type_util.cc57 DataLayout data_layout) { in ToTensorStorageType()
Dgpu_api_delegate.cc62 DataLayout ToDataLayoutFromTFL(TfLiteGpuDataLayout data_layout) { in ToDataLayoutFromTFL()
223 DataType data_type, DataLayout data_layout) { in BindGlBufferToTensor()
372 TfLiteType data_type, TfLiteGpuDataLayout data_layout) { in TfLiteGpuDelegateBindGlBufferToTensor()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsimple_orc_jit.h66 const llvm::DataLayout& data_layout() const { return data_layout_; } in data_layout() function
Dvector_support_library.cc152 const llvm::DataLayout& data_layout = in IntegerTypeForFloatSize() local
/external/llvm/bindings/ocaml/executionengine/
Dllvm_executionengine.mli71 val data_layout : llexecutionengine -> Llvm_target.DataLayout.t val
/external/swiftshader/third_party/llvm-7.0/llvm/bindings/ocaml/executionengine/
Dllvm_executionengine.mli71 val data_layout : llexecutionengine -> Llvm_target.DataLayout.t val
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto84 DataLayout data_layout = 3; field
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_init.c295 char *data_layout; in init_gallivm_engine() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc111 const char* target_triple, const char* data_layout) in GpuCompiler()
/external/swiftshader/third_party/llvm-7.0/llvm/bindings/ocaml/target/
Dllvm_target.mli198 val data_layout : t -> DataLayout.t val
/external/llvm/bindings/ocaml/target/
Dllvm_target.mli198 val data_layout : t -> DataLayout.t val
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dapi.h142 DataLayout data_layout = DataLayout::UNKNOWN; member
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc491 int64 ByteSizeOf(const Shape& shape, const llvm::DataLayout& data_layout) { in ByteSizeOf()
/external/llvm/bindings/ocaml/llvm/
Dllvm.mli471 val data_layout: llmodule -> string val
/external/swiftshader/third_party/llvm-7.0/llvm/bindings/ocaml/llvm/
Dllvm.mli492 val data_layout: llmodule -> string val
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc888 dnn::DataLayout data_layout = dnn::DataLayout::kBatchDepthYX) { in ToCudnnDataType()
917 dnn::DataLayout data_layout = dnn::DataLayout::kBatchDepthYX) { in GetCudnnDataType()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1579 dnn::DataLayout data_layout = dnn::DataLayout::kBatchDepthYX) { in ToMIOpenDataType()