Home
last modified time | relevance | path

Searched refs:cond_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dwhere.cc33 const TfLiteTensor* cond_tensor, in ResizeOutputTensor() argument
38 const RuntimeShape& cond_shape = GetTensorShape(cond_tensor); in ResizeOutputTensor()
41 const bool* cond_data = GetTensorData<bool>(cond_tensor); in ResizeOutputTensor()
59 const TfLiteTensor* cond_tensor; in Prepare() local
61 &cond_tensor)); in Prepare()
66 if (cond_tensor->type != kTfLiteBool) { in Prepare()
69 TfLiteTypeGetName(cond_tensor->type)); in Prepare()
79 if (!IsConstantTensor(cond_tensor)) { in Prepare()
83 return ResizeOutputTensor(context, cond_tensor, output); in Prepare()
87 const TfLiteTensor* cond_tensor; in Eval() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_gpu_select.cu.cc30 typename TTypes<bool, NDIMS>::ConstTensor cond_tensor, in operator ()()
36 output_tensor.device(d) = cond_tensor.broadcast(cond_bcast) in operator ()()
Dcwise_op_select.cc403 typename TTypes<bool, NDIMS>::ConstTensor cond_tensor, in operator ()()
409 output_tensor.device(d) = cond_tensor.broadcast(cond_bcast) in operator ()()
Dcwise_ops.h1357 typename TTypes<bool, NDIMS>::ConstTensor cond_tensor,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_helpers.inc219 auto cond_tensor = cond.getType().cast<RankedTensorType>();
222 if (cond_tensor.getRank() != 1 || data_rank <= 1) {
231 shape[0] = cond_tensor.getShape().front();
Dtf_ops_n_z.cc781 auto cond_tensor = op.condition().getType().dyn_cast<RankedTensorType>(); in Verify() local
782 if (!cond_tensor) return success(); in Verify()
783 auto cond_rank = cond_tensor.getRank(); in Verify()
788 auto cond_shape = cond_tensor.getShape().front(); in Verify()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py4710 cond_tensor = []
4713 if not cond_tensor:
4714 cond_tensor.append(constant_op.constant(1))
4715 return cond_tensor[0]
4719 return cond_tensor[0]
4764 cond_tensor = self._getCondTensor()
4765 math_ops.add(1, cond_tensor)
4770 cond_tensor = []
4773 if not cond_tensor:
4774 cond_tensor.append(constant_op.constant(1))
[all …]