HomeSort by relevance Sort by last modified time
    Searched refs:cond_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
where.cc 31 const TfLiteTensor* cond_tensor,
36 const RuntimeShape& cond_shape = GetTensorShape(cond_tensor);
39 const bool* cond_data = GetTensorData<bool>(cond_tensor);
57 const TfLiteTensor* cond_tensor = local
61 if (cond_tensor->type != kTfLiteBool) {
64 TfLiteTypeGetName(cond_tensor->type));
73 if (!IsConstantTensor(cond_tensor)) {
77 return ResizeOutputTensor(context, cond_tensor, output);
81 const TfLiteTensor* cond_tensor = local
87 ResizeOutputTensor(context, cond_tensor, output))
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
control_flow_ops_py_test.py     [all...]

Completed in 3465 milliseconds