| /external/tensorflow/tensorflow/core/kernels/ |
| dynamic_partition_op.cc | 129 out_flat; variable
130 out_flat.reserve(num_partitions_);
132 out_flat.push_back(outputs[p]->flat_outer_dims<T>());
148 OP_REQUIRES(c, FastBoundsCheck(oi, out_flat[p].dimension(0)),
153 out_flat[p].slice(out_indices, sizes) =
|
| gather_op.cc | 108 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); variable
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
|
| sparse_reduce_op.cc | 183 auto out_flat = out_values->flat<T>(); variable
184 out_flat.setZero();
221 out_flat(idx) = reduced_val();
289 auto out_flat = out_values_t->flat<T>(); variable
306 out_flat(i) = reduced_val();
|
| cwise_ops_common.h | 105 auto out_flat = out->flat<Tout>(); variable
109 eigen_device, out_flat, in0.template flat<Tin>(),
114 eigen_device, out_flat, in0.template scalar<Tin>(),
118 eigen_device, out_flat, in0.template flat<Tin>(),
216 auto out_flat = out->flat<Tout>(); variable
217 functor::SimpleBinaryFunctor<Device, Functor>()(eigen_device, out_flat,
|
| pooling_ops_common.h | 521 auto out_flat = output->flat<T>(); local
523 auto shard = [¶ms, &in_flat, &out_flat](int64 start, int64 limit) {
535 out_flat.data() + start * output_image_size, params.depth,
|
| resource_variable_ops.cc | 521 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); variable
524 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
|
| /external/tensorflow/tensorflow/contrib/kfac/python/ops/ |
| fisher_blocks.py | 217 out_flat = self._factor.left_multiply_inverse(
219 return utils.column_to_tensors(vector, out_flat)
223 out_flat = self._factor.left_multiply(
225 return utils.column_to_tensors(vector, out_flat)
281 out_flat = self._factor.left_multiply_inverse(
283 print("out_flat: %s" % out_flat)
284 return utils.column_to_tensors(vector, out_flat)
288 out_flat = self._factor.left_multiply(
290 return utils.column_to_tensors(vector, out_flat)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| dot_operation_test.cc | 507 auto out_flat = builder.ConcatInDim(out_slices, 0); local
508 builder.Reshape(out_flat, {0, 1, 2}, {2, 2, 2, 2});
|
