| 517 | }; |
| 518 | |
| 519 | TensorObjectDef TensorToDef(const Tensor& tensor) { |
| 520 | TensorObjectDef def; |
| 521 | def.dimensions.b = 1; |
| 522 | def.dimensions.h = tensor.Height(); |
| 523 | def.dimensions.w = tensor.Width(); |
| 524 | def.dimensions.c = tensor.Channels(); |
| 525 | def.object_def.data_layout = ToDataLayout(tensor.StorageType()); |
| 526 | def.object_def.data_type = tensor.DataType(); |
| 527 | def.object_def.object_type = ToObjectType(tensor.StorageType()); |
| 528 | def.object_def.user_provided = false; |
| 529 | return def; |
| 530 | } |
| 531 | |
| 532 | class InferenceBuilderImpl : public InferenceBuilder { |
| 533 | public: |
no test coverage detected