Home
last modified time | relevance | path

Searched refs:else_tensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_gpu_select.cu.cc32 typename TTypes<T, NDIMS>::ConstTensor else_tensor, in operator ()()
38 else_tensor.broadcast(else_bcast)); in operator ()()
Dcwise_op_select.cc405 typename TTypes<T, NDIMS>::ConstTensor else_tensor, in operator ()()
411 else_tensor.broadcast(else_bcast)); in operator ()()
Dcwise_ops.h1359 typename TTypes<T, NDIMS>::ConstTensor else_tensor,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc749 auto else_tensor = op.e().getType().cast<TensorType>(); in Verify() local
751 if (!AreCastCompatible({then_tensor, else_tensor})) in Verify()
760 bool else_has_rank = else_tensor.hasRank(); in Verify()
765 if (else_tensor.getRank() > 0) in Verify()
767 static_cast<int>(else_tensor.getShape().front()), data_first_dim); in Verify()
773 data_rank = else_tensor.getRank(); in Verify()
774 if (else_tensor.getRank() > 0) in Verify()
775 data_first_dim = else_tensor.getShape().front(); in Verify()