Searched refs:else_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 33 typename TTypes<T, NDIMS>::ConstTensor else_tensor, in operator ()()
39 else_tensor.broadcast(else_bcast)); in operator ()()
|
| D | cwise_op_select.cc | 421 typename TTypes<T, NDIMS>::ConstTensor else_tensor, in operator ()()
427 else_tensor.broadcast(else_bcast)); in operator ()()
|
| D | cwise_ops.h | 1385 typename TTypes<T, NDIMS>::ConstTensor else_tensor,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 916 auto else_tensor = op.e().getType().cast<TensorType>(); in verify() local
918 if (!AreCastCompatible({then_tensor, else_tensor})) in verify()
927 bool else_has_rank = else_tensor.hasRank(); in verify()
932 if (else_tensor.getRank() > 0) in verify()
934 static_cast<int>(else_tensor.getShape().front()), data_first_dim); in verify()
940 data_rank = else_tensor.getRank(); in verify()
941 if (else_tensor.getRank() > 0) in verify()
942 data_first_dim = else_tensor.getShape().front(); in verify()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 2023 Tensor<BHWC, DataType::FLOAT32> else_tensor; in Parse() local
2036 RETURN_IF_ERROR(reader->ReadTensor(2, &else_tensor)); in Parse()
2041 else_tensor.shape = BHWC(1, 1, 1, 1); in Parse()
2042 else_tensor.data.push_back(else_scalar_tensor.data[0]); in Parse()
2053 RETURN_IF_ERROR(NewConstNode(else_tensor, graph, &else_value)); in Parse()
|