Home
last modified time | relevance | path

Searched refs:output_def (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc42 const TensorObjectDef& output_def,
121 const TensorObjectDef& output_def, in Init() argument
134 output_def.object_def.object_type, output_def.object_def.data_layout); in Init()
135 dst_tensor_descriptor_.data_type = output_def.object_def.data_type; in Init()
142 output_def.object_def.data_type == DataType::FLOAT16; in Init()
144 ToCLDataType(output_def.object_def.data_type); in Init()
209 const TensorObjectDef& output_def, in Init() argument
222 output_def.object_def.data_type == DataType::FLOAT16; in Init()
228 ToCLDataType(output_def.object_def.data_type); in Init()
292 const TensorObjectDef& output_def) const { in GetFromBhwcKernel()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc53 const TensorObjectDef& output_def) = 0;
107 const TensorObjectDef& output_def) final { in Init() argument
108 shape_ = BHWC(output_def.dimensions.b, output_def.dimensions.h, in Init()
109 output_def.dimensions.w, output_def.dimensions.c); in Init()
195 const TensorObjectDef& output_def) final { in Init() argument
196 shape_ = BHWC(output_def.dimensions.b, output_def.dimensions.h, in Init()
197 output_def.dimensions.w, output_def.dimensions.c); in Init()
350 const auto& output_def = output.object_def; in IsSupported() local
352 (TrivialCopier::IsSupported(input_def, output_def) || in IsSupported()
353 CpuCopier::IsSupported(input_def, output_def) || in IsSupported()
[all …]
Dconverter_test.cc76 TensorObjectDef output_def = input_def; in RunFromTensorTest() local
77 output_def.object_def.data_layout = DataLayout::BHWC; in RunFromTensorTest()
79 RETURN_IF_ERROR(builder->MakeConverter(input_def, output_def, &converter)); in RunFromTensorTest()
133 TensorObjectDef output_def = input_def; in RunToTensorTest() local
134 output_def.object_def.data_layout = DataLayout::DHWC4; in RunToTensorTest()
136 RETURN_IF_ERROR(builder->MakeConverter(input_def, output_def, &converter)); in RunToTensorTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgl_interop.h156 const TensorObjectDef& output_def, Environment* environment);
Dgl_interop.cc264 const TensorObjectDef& output_def, in GlClBufferCopier() argument
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc320 ArgumentSpec CreateOutput(const OpDef_ArgDef& output_def, in CreateOutput() argument
324 Type type = type_resolver->TypeOf(output_def, &iterable); in CreateOutput()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc403 for (const auto& output_def : output_defs) { in Initialize() local
404 output_to_cpu_ |= output_def.external_def.object_def.object_type == in Initialize()
/external/llvm-project/lld/MinGW/
DOptions.td74 defm output_def: Eq<"output-def", "Output def file">;
/external/llvm-project/lld/COFF/
DOptions.td217 def output_def : Joined<["/", "-", "/?", "-?"], "output-def:">;