Searched defs:parameter_shape (Results 1 – 9 of 9) sorted by relevance
72 const Shape& parameter_shape(int64 param_no) const { in parameter_shape() function
344 const Shape& parameter_shape = in CreateFromProto() local
646 const Shape& parameter_shape = to_apply.parameters(i); in SameElementTypesForOperandsAndToApplyParameters() local
1187 const Shape& parameter_shape = to_apply.parameters(i); in InferMapShape() local
374 shape_inference::ShapeHandle parameter_shape; in operator ()() local
308 for (auto& parameter_shape : program_shape.parameters()) { in HasLayout() local
224 bool InputShapeMatches(const xla::Shape& parameter_shape, in InputShapeMatches()
5209 Shape parameter_shape = TypeToShape(parameter.getType()); in NumInputsInvolveInOnlyElementwiseOps() local5222 Shape parameter_shape = TypeToShape(parameter.getType()); in NumInputsWithMoreElementsThan() local
537 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias() local