Searched defs:parameter_shape (Results 1 – 9 of 9) sorted by relevance
72 const Shape& parameter_shape(int64_t param_no) const { in parameter_shape() function
351 const Shape& parameter_shape = in CreateFromProto() local
983 const Shape& parameter_shape = to_apply.parameters(i); in SameElementTypesForOperandsAndToApplyParameters() local
1176 const Shape& parameter_shape = to_apply.parameters(i); in InferMapShape() local
432 shape_inference::ShapeHandle parameter_shape; in operator ()() local
308 for (auto& parameter_shape : program_shape.parameters()) { in HasLayout() local
225 bool InputShapeMatches(const xla::Shape& parameter_shape, in InputShapeMatches()
4737 Shape parameter_shape = GetShape(parameter); in NumInputsInvolveInOnlyElementwiseOps() local4750 Shape parameter_shape = GetShape(parameter); in NumInputsWithMoreElementsThan() local
547 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias() local