| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_clip.cc | 55 auto out_flat = out->flat<T>(); in Compute() local
61 out_flat); in Compute()
64 out_flat); in Compute()
69 out_flat); in Compute()
72 out_flat); in Compute()
95 typename TTypes<T>::Flat& out_flat) const { in operator ()()
96 out_flat = in0_flat.unaryExpr(UnaryClipFunc<T>(in1_flat(0), in2_flat(0))); in operator ()()
114 typename TTypes<T>::Flat& out_flat) const { in operator ()()
115 out_flat = in operator ()()
134 typename TTypes<T>::Flat& out_flat) const { in operator ()()
[all …]
|
| D | cwise_op_clip_gpu.cu.cc | 62 typename TTypes<T>::Flat &out_flat) const { in operator ()()
68 in2_flat.data(), out_flat.data())); in operator ()()
78 typename TTypes<T>::Flat &out_flat) const { in operator ()()
84 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
94 typename TTypes<T>::Flat &out_flat) const { in operator ()()
100 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
110 typename TTypes<T>::Flat &out_flat) const { in operator ()()
111 out_flat.device(d) = in0_flat.cwiseMin(in2_flat).cwiseMax(in1_flat); in operator ()()
|
| D | cwise_op_clip.h | 29 typename TTypes<T>::Flat &out_flat) const;
38 typename TTypes<T>::Flat &out_flat) const;
47 typename TTypes<T>::Flat &out_flat) const;
56 typename TTypes<T>::Flat &out_flat) const;
|
| D | dynamic_partition_op.cc | 129 out_flat; in Compute() local
130 out_flat.reserve(num_partitions_); in Compute()
132 out_flat.push_back(outputs[p]->flat_outer_dims<T>()); in Compute()
148 OP_REQUIRES(c, FastBoundsCheck(oi, out_flat[p].dimension(0)), in Compute()
153 out_flat[p].slice(out_indices, sizes) = in Compute()
|
| D | sparse_reduce_op.cc | 185 auto out_flat = out_values->flat<T>(); in Compute() local
186 out_flat.setZero(); in Compute()
223 out_flat(idx) = reduced_val(); in Compute()
293 auto out_flat = out_values_t->flat<T>(); in Compute() local
310 out_flat(i) = reduced_val(); in Compute()
|
| D | gather_op.cc | 108 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); in Compute() local
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
|
| D | unary_ops_composition.cc | 114 OutputBuffer out_flat = out->flat<T>(); in Compute() local
117 auto compute_fn = [this, &in_flat, &out_flat, &num_fns](int64 begin, in Compute()
121 const InputBuffer scratch_slice(out_flat.data() + begin, len); in Compute()
122 OutputBuffer out_slice(out_flat.data() + begin, len); in Compute()
|
| D | cwise_ops_common.h | 107 auto out_flat = out->flat<Tout>(); in Compute() local
111 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
116 eigen_device, out_flat, in0.template scalar<Tin>(), in Compute()
120 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
218 auto out_flat = out->flat<Tout>(); in Compute() local
219 functor::SimpleBinaryFunctor<Device, Functor>()(eigen_device, out_flat, in Compute()
|
| D | pooling_ops_common.h | 521 auto out_flat = output->flat<T>();
523 auto shard = [¶ms, &in_flat, &out_flat](int64 start, int64 limit) {
535 out_flat.data() + start * output_image_size, params.depth,
|
| D | resource_variable_ops.cc | 691 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); in Compute() local
694 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 574 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST() local
575 Reshape(out_flat, {0, 1, 2}, {2, 2, 2, 2}); in XLA_TYPED_TEST()
|