HomeSort by relevance Sort by last modified time
    Searched refs:smaller_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.cc 784 BinaryOperation operation, const Shape& smaller_shape,
787 if (broadcast_dimensions.empty() && !ShapeUtil::IsScalar(smaller_shape)) {
792 ShapeUtil::HumanString(smaller_shape).c_str(),
794 } else if (broadcast_dimensions.size() != ShapeUtil::Rank(smaller_shape)) {
800 ShapeUtil::Rank(smaller_shape), broadcast_dimensions.size());
810 // smaller_shape larger_shape broadcast_dimensions output_shape
926 const Shape& smaller_shape = local
    [all...]
shape_inference.h 291 // (for example ComputationBuilder::AddInDim). smaller_shape must be a
293 // smaller_shape is broadcast to.
295 BinaryOperation operation, const Shape& smaller_shape,
  /external/tensorflow/tensorflow/python/ops/linalg/
linear_operator_test_util.py 664 smaller_shape = batch_shape + [m, n - 1]
675 smaller_shape, mean=0.0, stddev=stddev_mat, dtype=dtype, seed=seed)

Completed in 375 milliseconds