| /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 | 70 typename TTypes<T>::Flat &out_flat) const { in operator ()()
76 in2_flat.data(), out_flat.data())); in operator ()()
86 typename TTypes<T>::Flat &out_flat) const { in operator ()()
92 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
102 typename TTypes<T>::Flat &out_flat) const { in operator ()()
108 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
118 typename TTypes<T>::Flat &out_flat) const { in operator ()()
119 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 | gather_op.cc | 167 auto out_flat = out->shaped<T, 4>( in Compute() local
171 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
175 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); in Compute() local
178 bad_i = functor(c, params_flat, indices_flat, out_flat); 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 | 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 | 168 auto out_flat = out->flat<Tout>(); in Compute() local
172 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
177 eigen_device, out_flat, in0.template scalar<Tin>(), in Compute()
181 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
279 auto out_flat = out->flat<Tout>(); in Compute() local
280 functor::SimpleBinaryFunctor<Device, Functor>()(eigen_device, out_flat, in Compute()
|
| D | ragged_tensor_to_variant_op.cc | 236 Piece out_flat = in Compute() local
238 ConcatCPU<VALUE_TYPE>(context->device(), pieces, &out_flat); in Compute()
|
| D | pooling_ops_common.h | 573 auto out_flat = output->flat<T>();
575 auto shard = [¶ms, &in_flat, &out_flat](int64 start, int64 limit) {
587 out_flat.data() + start * output_image_size, params.depth,
|
| D | resource_variable_ops.cc | 695 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); in Compute() local
698 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 618 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST() local
619 Reshape(out_flat, {0, 1, 2}, {2, 2, 2, 2}); in XLA_TYPED_TEST()
|