Searched defs:tensor_rank (Results 1 – 6 of 6) sorted by relevance
34 int tensor_rank, in GetOutputLayout()87 const int tensor_rank = ValueRank(scatter_op.tensor()); in TensorScatterOpExpand() local159 const int tensor_rank = ValueRank(scatter_op.tensor()); in TensorScatterOpComputeLayoutForward() local187 const int tensor_rank = ValueRank(scatter_op.tensor()); in TensorScatterOpComputeLayoutBackward() local
47 int64_t tensor_rank = ValueRank(cumsum.x()); in GetAxisDimension() local
387 def tensor_rank(self, name="tensor_rank"): member in LinearOperator
513 uint32_t tensor_rank = static_cast<uint32_t>(tensor->dims->size); in ConvertTensorTypeToNNType() local1532 const uint32_t tensor_rank = static_cast<uint32_t>(dims->size); in AddNewInputConstantTensor() local1840 uint32_t tensor_rank = static_cast<uint32_t>(tensor->dims->size); in AddTensor() local
653 int32_t tensor_rank = result_type.getShape().size(); in convertConcatV2Op() local
4510 int64_t tensor_rank = tensor_ty.getRank(); in matchAndRewrite() local