Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_gpu_select.cu.cc31 typename TTypes<T, NDIMS>::ConstTensor then_tensor, in operator ()()
37 .select(then_tensor.broadcast(then_bcast), in operator ()()
Dcwise_op_select.cc404 typename TTypes<T, NDIMS>::ConstTensor then_tensor, in operator ()()
410 .select(then_tensor.broadcast(then_bcast), in operator ()()
Dcwise_ops.h1358 typename TTypes<T, NDIMS>::ConstTensor then_tensor,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc748 auto then_tensor = op.t().getType().cast<TensorType>(); in Verify() local
751 if (!AreCastCompatible({then_tensor, else_tensor})) in Verify()
759 bool then_has_rank = then_tensor.hasRank(); in Verify()
762 data_rank = then_tensor.getRank(); in Verify()
763 if (then_tensor.getRank() > 0) in Verify()
764 data_first_dim = then_tensor.getShape().front(); in Verify()
769 data_rank = then_tensor.getRank(); in Verify()
770 if (then_tensor.getRank() > 0) in Verify()
771 data_first_dim = then_tensor.getShape().front(); in Verify()