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.cc920 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim() local
922 in_dim_size[broadcast_dimensions[i]] = operand_shape->dimensions(i); in BroadcastInDim()
925 ShapeUtil::MakeShape(operand_shape->element_type(), in_dim_size); in BroadcastInDim()