Searched refs:scalar_indices (Results 1 – 2 of 2) sorted by relevance
32 absl::InlinedVector<xla::XlaOp, 4> scalar_indices; in SliceVector() local33 scalar_indices.reserve(rank); in SliceVector()35 scalar_indices.push_back( in SliceVector()37 return scalar_indices; in SliceVector()
137 absl::InlinedVector<xla::XlaOp, 4> scalar_indices; in Compile() local138 scalar_indices.reserve(input_dims); in Compile()141 scalar_indices.push_back( in Compile()146 xla::DynamicSlice(ctx->Input(0), scalar_indices, size)); in Compile()165 auto sliced = xla::DynamicSlice(padded_input, scalar_indices, in Compile()