Searched defs:parameter_shape (Results 1 – 8 of 8) sorted by relevance
72 const Shape& parameter_shape(int64 param_no) const { in parameter_shape() function
286 const Shape& parameter_shape = in CreateFromProto() local
428 const Shape& parameter_shape = to_apply.parameters(i); in SameElementTypesForOperandsAndToApplyParameters() local
1143 const Shape& parameter_shape = to_apply.parameters(i); in InferMapShape() local
177 shape_inference::ShapeHandle parameter_shape; in RegisterPerTableLoadOpsForAlgorithmBody() local
319 for (auto& parameter_shape : program_shape.parameters()) { in HasLayout() local
3574 const Shape& parameter_shape = parameter->shape(); in NumInputsInvolveInOnlyElementwiseOps() local
404 const Shape& parameter_shape = program_shape.parameters(alias.param_number); in PopulateInputOutputAlias() local