Searched defs:image_shape (Results 1 – 11 of 11) sorted by relevance
28 …Resize2D(const SymbolPtr &image_shape, const SymbolPtr &size_array) : InferShapeOp({image_shape, s… in Resize2D()37 auto image_shape = input_as<ListSymbol>(kIndex0); in Eval() local
109 def construct(self, feats, gts, image_shape): argument142 def predict(self, feats, image_shape): argument
23 const auto &image_shape = input_args[1]->GetShape(); in InferShape() local
32 auto image_shape = input_args[0]->GetShape(); in Resize2DInferShape() local
36 TensorShape image_shape = inputs[0]; in OutputShape() local
70 TensorShape image_shape = image->shape(); in ValidateCutMixBatch() local118 TensorShape image_shape = image->shape(); in ComputeImage() local193 TensorShape image_shape = input.at(0)->shape(); in Compute() local
86 std::vector<int64_t> image_shape = input.at(0)->shape().AsVector(); in Compute() local
39 ShapeVector image_shape; member
38 ShapeVector image_shape; member
127 auto image_shape = EigenTensor(image_bcast_, nullptr).AsEigenDSizes<kMaxDims>(); in LaunchKernel() local