Searched defs:cond_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | where.cc | 35 const TfLiteTensor* cond_tensor, in ResizeOutputTensor()
59 const TfLiteTensor* cond_tensor, in PrepareOutput()
78 const TfLiteTensor* cond_tensor; in Prepare() local
109 const TfLiteTensor* cond_tensor; in Eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | select_v2_test_util.cc | 33 std::vector<float> SetUpData(Tensor<BHWC, cond_type>& cond_tensor, in SetUpData()
111 const Tensor<BHWC, cond_type>& cond_tensor, in RunSelectV2()
147 Tensor<BHWC, cond_type> cond_tensor; in SelectV2TestTemplate() local
175 Tensor<BHWC, cond_type> cond_tensor; in SelectV2BatchTestTemplate() local
205 Tensor<BHWC, cond_type> cond_tensor; in SelectV2BroadcastFalseTestTemplate() local
235 Tensor<BHWC, cond_type> cond_tensor; in SelectV2BroadcastTrueTestTemplate() local
265 Tensor<BHWC, cond_type> cond_tensor; in SelectV2BroadcastBothTestTemplate() local
293 Tensor<BHWC, cond_type> cond_tensor; in SelectV2ChannelsTestTemplate() local
322 Tensor<BHWC, cond_type> cond_tensor; in SelectV2ChannelsBatchTestTemplate() local
353 Tensor<BHWC, cond_type> cond_tensor; in SelectV2ChannelsBroadcastFalseTestTemplate() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/ |
| D | multiplex_2_kernel.h | 53 const auto& cond_tensor = ctx->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_1/ |
| D | multiplex_1_kernel.cc | 40 const auto& cond_tensor = ctx->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_4/ |
| D | multiplex_4_kernel.cc | 98 const auto& cond_tensor = ctx->input(list_index); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 948 auto cond_tensor = op.condition().getType().dyn_cast<RankedTensorType>(); in verify() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 1999 const TfLiteTensor* cond_tensor = reader->GetInputTensor(0); in Parse() local
|