Searched defs:current_shape (Results 1 – 5 of 5) sorted by relevance
86 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local126 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute() local
100 auto current_shape = shapes[i]; in IncludesShapes() local
227 std::vector<int> current_shape = imagined_original_shape; in Run() local
1330 const xla::HloInstruction* instr, const xla::Shape& current_shape, in GetOrCreateArrayView()1399 const HloInstruction* instr, const Shape& current_shape, in GetOrCreateViewImpl()
68 SmallVector<int64_t, 4> current_shape; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local