Searched refs:SpanToVector (Results 1 – 12 of 12) sorted by relevance
233 std::vector<int64> dimensions = xla::SpanToVector(shape.dimensions()); in GetTensorListShapeFromElementTensorListShape()258 std::vector<int64> dimensions = xla::SpanToVector(element_shape.dimensions()); in GetTensorListShapeFromElementShape()376 xla::SpanToVector(element_part_shape.dimensions()); in ExecuteTensorListPushBack()393 xla::SpanToVector(element_shape.dimensions()); in ExecuteTensorListPushBack()441 xla::SpanToVector(list_part_shape.dimensions()); in ExecuteTensorListPopBack()480 xla::SpanToVector(element_shape.dimensions()); in ExecuteTensorListSetItem()521 std::vector<int64> slice_shape = xla::SpanToVector(buffer_shape.dimensions()); in ExecuteTensorListGetItem()
161 xla::SpanToVector(builder->GetShape(conv).ValueOrDie().dimensions()); in Compile()
488 xla::SpanToVector(element_shape.dimensions()); in Compile()534 xla::SpanToVector(element_shape.dimensions()); in Compile()
63 xla::SpanToVector(input_shape.dimensions()); in Expand()
87 std::vector<int64> reshape_dims = SpanToVector(lhs->shape().dimensions()); in ConvertBatchGroupedToFeatureGroupedConvolution()567 std::vector<int64> reshape_dims = SpanToVector(rhs->shape().dimensions()); in MatchBackwardInput()589 SpanToVector(rhs->shape().dimensions()); in MatchBackwardInput()
967 SpanToVector(scalar_indexed_const->output_dims()), in ComputeArrayForElementwiseUnaryOp()1063 SpanToVector(lhs->output_dims()), shape); in ComputeArrayForDotWithIndexedLhs()1099 SpanToVector(rhs->output_dims()), shape); in ComputeArrayForDotWithIndexedRhs()
2237 SpanToVector(constant->shape().dimensions()); in OptimizeDotOfReorderContractingDims()2259 SpanToVector(rhs_reshape->shape().dimensions()); in OptimizeDotOfReorderContractingDims()
3573 SpanToVector(scatter_indices_shape.dimensions()); in InferScatterShape()
278 std::vector<int64> slice_sizes = SpanToVector(input_shape.dimensions()); in TorchIndexSelect()
511 std::vector<T> SpanToVector(absl::Span<const T> slice) { in SpanToVector() function
1434 SpanToVector(simple_output_shape->layout().minor_to_major()); in AlignLayouts()
479 return xla::SpanToVector(absl::get<xla::Shape>(shape).dimensions()); in DimensionSizes()