Searched refs:input_as_matrix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_variant_op.cc | 84 const auto& input_as_matrix = input.flat_inner_dims<Variant, 2>(); in Compute() local
104 input_as_matrix(0, 1), input_as_matrix(0, 2), 0, in Compute()
131 input_as_matrix(i, 1), input_as_matrix(i, 2), in Compute()
176 input_as_matrix(i, 0), input_as_matrix(i, 1), in Compute()
|
| D | spectrogram_op.cc | 50 const auto input_as_matrix = input.matrix<float>(); in Compute() local
78 input_for_channel[i] = input_as_matrix(i, channel); in Compute()
|