OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:data_flat
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc
177
CudaDeviceArrayOnHost<const T*>
data_flat
(c, data_elements_size);
179
OP_REQUIRES_OK(c,
data_flat
.Init());
185
//
data_flat
index
197
data_flat
.Set(
205
OP_REQUIRES_OK(c,
data_flat
.Finalize());
209
indices_flat.data(),
data_flat
.data(), output);
246
auto
data_flat
=
251
const T* data_base = &
data_flat
(0, 0);
271
data_flat
.slice(data_indices, sizes);
dynamic_partition_op.cc
104
const auto
data_flat
= data->flat<T>();
variable
122
out_vec[p](oi) =
data_flat
(i);
137
const auto
data_flat
= data->shaped<T, 2>({N, slice_size});
variable
154
data_flat
.slice(data_indices, sizes);
segment_reduction_ops.cc
379
auto
data_flat
= typename TTypes<T, 2>::ConstTensor(data, N, data_size / N);
local
389
reduction(
data_flat
.template chip<0>(i), output.template chip<0>(j));
[
all
...]
Completed in 119 milliseconds