Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getMaxNumElements (0.25 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

          return list.emitOpError("unknown tensor list element shape");
        }
      }
      if (failed(cutil::CreateInitBufferValue(
              element_shape, list.getMaxNumElements(), list, list.getElementDtype(),
              builder, &buffer))) {
        return failure();
      }
      Value size = cutil::GetR1Const({0LL}, builder, list.getLoc());
      list.getHandle().replaceAllUsesWith(buffer);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

          !IsOfRankOrUnranked(op.getElementShape(), 1)) {
        return op.emitOpError("requires element_shape operand to be 0D/1D tensor");
      }
    
      if (!IsOfRankOrUnranked(op.getMaxNumElements(), 0)) {
        return op.emitOpError("requires max_num_elements operand to be 0D tensor");
      }
      return success();
    }
    
    //===----------------------------------------------------------------------===//
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top