Searched defs:input_shape_dims (Results 1 – 3 of 3) sorted by relevance
33 PrimitiveType primitive_type, absl::Span<const int64_t> input_shape_dims, in CreateModuleWithProgramShape()48 PrimitiveType primitive_type, absl::Span<const int64_t> input_shape_dims, in CreateModuleWithProgramShape()
84 absl::Span<int64_t const> input_shape_dims = in RewriteReduction() local
1263 absl::Span<int64_t const> input_shape_dims(shape.dimensions()); in CollapseFirstNDims() local