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

  /frameworks/ml/nn/common/operations/
SimpleMath.cpp 53 bool needBroadcast = !SameShape(shape1, shape2);
83 bool needBroadcast = !SameShape(shape1, shape2);
153 bool needBroadcast = !SameShape(shape1, shape2);
237 bool needBroadcast = !SameShape(shape1, shape2);
262 bool needBroadcast = !SameShape(shape1, shape2);
  /frameworks/ml/nn/common/include/
OperationsUtils.h 57 bool SameShape(const Shape& in1, const Shape& in2);
  /external/tensorflow/tensorflow/core/kernels/
sparse_dense_binary_op_shared_test.cc 93 TEST_F(SparseDenseCDivTest, SameShape) {
  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier.cc 237 bool SameShape(const HloInstruction* lhs, const HloInstruction* rhs) const;
318 bool AlgebraicSimplifierVisitor::SameShape(const HloInstruction* lhs,
343 if (!SameShape(old_instruction, new_instruction)) {
    [all...]
  /frameworks/ml/nn/common/
OperationsUtils.cpp 28 bool SameShape(const Shape& in1, const Shape& in2) {
192 if (SameShape(in1, in2)) {
    [all...]

Completed in 153 milliseconds