Home
last modified time | relevance | path

Searched refs:in_dim_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc693 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim() local
695 in_dim_size[broadcast_dimensions[i]] = operand_shape.dimensions(i); in BroadcastInDim()
698 ShapeUtil::MakeShape(operand_shape.element_type(), in_dim_size); in BroadcastInDim()