Searched refs:out_dim_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 672 const XlaOp& operand, const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
679 ShapeUtil::MakeShape(operand_shape.element_type(), out_dim_size); in BroadcastInDim()
682 broadcast_dimensions[i] > out_dim_size.size()) { in BroadcastInDim()
693 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim()
2844 const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
2846 return operand.builder()->BroadcastInDim(operand, out_dim_size, in BroadcastInDim()
|
| D | xla_builder.h | 333 const absl::Span<const int64> out_dim_size,
719 const XlaOp& operand, const absl::Span<const int64> out_dim_size,
1138 const absl::Span<const int64> out_dim_size,
|