Searched refs:object_def (Results 1 – 9 of 9) sorted by relevance
125 input_def.object_def.object_type, input_def.object_def.data_layout); in Init()126 src_tensor_descriptor_.data_type = input_def.object_def.data_type; in Init()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()141 input_def.object_def.data_type == DataType::FLOAT16 || in Init()142 output_def.object_def.data_type == DataType::FLOAT16; in Init()144 ToCLDataType(output_def.object_def.data_type); in Init()212 input_def.object_def.object_type, input_def.object_def.data_layout); in Init()215 tensor_descriptor_.data_type = input_def.object_def.data_type; in Init()221 input_def.object_def.data_type == DataType::FLOAT16 || in Init()[all …]
61 if (def.object_def.object_type != gpu::ObjectType::OPENGL_SSBO) { in MaybeAllocateGlBuffer()65 switch (def.object_def.data_type) { in MaybeAllocateGlBuffer()133 if (!def().external_def.object_def.user_provided) { in SetExternalObject()147 if (def().external_def.object_def.object_type == in SetExternalObject()164 const auto& external_def = def().external_def.object_def; in IsSameDef()165 const auto& internal_def = def().internal_def.object_def; in IsSameDef()179 const auto& external_def = def().external_def.object_def; in Init()225 if (d.object_def.user_provided) { in MaybeAllocateInternalObject()228 switch (d.object_def.object_type) { in MaybeAllocateInternalObject()245 switch (d.object_def.object_type) { in MaybeAllocateExternalObject()[all …]
66 if (!def().external_def.object_def.user_provided) { in SetExternalObject()96 auto object_type = def.external_def.object_def.object_type; in IsSupported()98 if (def.external_def.object_def.user_provided && in IsSupported()99 GlClBufferCopier::IsSupported(def.external_def.object_def, in IsSupported()100 def.internal_def.object_def)) { in IsSupported()135 if (!def().external_def.object_def.user_provided) { in SetExternalObject()151 if (def().external_def.object_def.user_provided && in Init()152 GlClBufferCopier::IsSupported(def().external_def.object_def, in Init()153 def().internal_def.object_def)) { in Init()160 if (def().external_def.object_def.user_provided && in Init()[all …]
241 def.object_def.data_type = data_type; in BindGlBufferToTensor()242 def.object_def.data_layout = data_layout; in BindGlBufferToTensor()243 def.object_def.object_type = ObjectType::OPENGL_SSBO; in BindGlBufferToTensor()244 def.object_def.user_provided = true; in BindGlBufferToTensor()254 return tensors_[index].second.object_def; in GetObjectDef()
268 NumElements(input_def) * SizeOf(input_def.object_def.data_type); in GlClBufferCopier()
72 input_def.object_def.data_type = DataType::FLOAT32; in RunFromTensorTest()73 input_def.object_def.data_layout = DataLayout::DHWC4; in RunFromTensorTest()74 input_def.object_def.object_type = ObjectType::OPENGL_SSBO; in RunFromTensorTest()77 output_def.object_def.data_layout = DataLayout::BHWC; in RunFromTensorTest()129 input_def.object_def.data_type = DataType::FLOAT32; in RunToTensorTest()130 input_def.object_def.data_layout = DataLayout::BHWC; in RunToTensorTest()131 input_def.object_def.object_type = ObjectType::OPENGL_SSBO; in RunToTensorTest()134 output_def.object_def.data_layout = DataLayout::DHWC4; in RunToTensorTest()
349 const auto& input_def = input.object_def; in IsSupported()350 const auto& output_def = output.object_def; in IsSupported()362 const auto& input_def = input.object_def; in MakeConverter()363 const auto& output_def = output.object_def; in MakeConverter()
73 bool IsValid(const TensorObjectDef& def) { return IsValid(def.object_def); } in IsValid()76 return GetType(object) == def.object_def.object_type && in IsValid()77 absl::visit(ObjectValidityChecker{def.object_def.data_type}, object); in IsValid()107 switch (def.object_def.data_layout) { in NumElements()
218 ObjectDef object_def; member221 return dimensions == other.dimensions && object_def == other.object_def;