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

  /external/tensorflow/tensorflow/core/kernels/
cwise_op_gpu_select.cu.cc 61 TTypes<bool>::ConstVec cond_vec,
64 const int batch = cond_vec.size();
80 // To32Bit(cond_vec)
86 cond_vec.reshape(reshape_dims)
cwise_op_select.cc 262 TTypes<bool>::ConstVec cond_vec,
265 const Eigen::DenseIndex batch = cond_vec.size();
279 cond_vec.reshape(reshape_dims)
290 TTypes<bool>::ConstVec cond_vec,
293 const size_t batch = cond_vec.size();
296 const bool* c = cond_vec.data();
cwise_ops.h     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
heap_simulator_test.cc 193 std::vector<HloInstruction*> cond_vec = {cond_param, slice, reshape, zero, local
199 schedule.set_sequence(cond_computation, cond_vec);
    [all...]

Completed in 437 milliseconds