Searched defs:input_flat (Results 1 – 14 of 14) sorted by relevance
133 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local139 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
39 const auto& input_flat = input_tensor->flat<string>(); in Compute() local81 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
43 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
36 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
243 const T* input_flat = input.template flat<T>().data(); in Compute() local299 const T* input_flat = input.template flat<T>().data(); in Compute() local
39 auto input_flat = input.flat<T1>(); in ReferenceBatchNorm() local102 auto input_flat = input.flat<T1>(); in FixedPointBatchNorm() local
89 auto input_flat = input.flat_outer_dims<T>(); in Compute() local654 auto input_flat = input.flat_outer_dims<T>(); in Compute() local753 int64 Reduce(const typename TTypes<T>::ConstMatrix& input_flat, in Reduce()1023 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
125 const auto input_flat = input.flat<string>(); in Compute() local
288 auto input_flat = input.flat<T>().data(); in Compute() local
379 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local409 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local440 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local471 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
238 TTypes<float, 1>::ConstTensor input_flat(&input(0, 0, 0), input.size()); in ReduceMeanAcrossImage() local
810 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local856 const auto& input_flat = input.flat<T1>(); in QuantizedAdd() local
217 auto input_flat = input_tensor.flat<T>(); in PopulateFromDenseGroup() local
39 const auto& input_flat = input_tensor->flat<string>(); in Compute() local