Searched defs:parameter_shape (Results 1 – 11 of 11) sorted by relevance
72 const Shape& parameter_shape(int64_t param_no) const { in parameter_shape() function
403 const Shape& parameter_shape = in CreateFromProto() local
766 const Shape& parameter_shape = instruction->operand(i)->shape(); in CreateAsyncInstructions() local
1186 const Shape& parameter_shape = to_apply.parameters(i); in InferMapShape() local
1009 const Shape& parameter_shape = to_apply.parameters(i); in SameElementTypesForOperandsAndToApplyParameters() local
432 shape_inference::ShapeHandle parameter_shape; in operator ()() local
260 ShapeHandle parameter_shape; in __anon5cd625660702() local
225 bool InputShapeMatches(const xla::Shape& parameter_shape, in InputShapeMatches()
372 for (auto& parameter_shape : program_shape.parameters()) { in HasLayout() local
4369 Shape parameter_shape = GetShape(parameter); in NumInputsInvolveInOnlyElementwiseOps() local4382 Shape parameter_shape = GetShape(parameter); in NumInputsWithMoreElementsThan() local
600 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias() local