| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| clustering_ops_test.cc | 58 points.flat<float>().setRandom();
59 sample_size.flat<int64>().setConstant(num_to_sample);
60 seed.flat<int64>().setConstant(12345);
61 num_retries_per_sample.flat<int64>().setConstant(retries_per_sample);
123 distances.flat<float>().setRandom();
124 seed.flat<int64>().setConstant(12345);
181 points.flat<float>().setRandom();
182 centers.flat<float>().setRandom();
183 top.flat<int64>().setConstant(k);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| fractional_avg_pool_op.cc | 119 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3],
122 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3],
131 auto output_row_seq_flat = output_row_seq_tensor->flat<int64>();
132 auto output_col_seq_flat = output_col_seq_tensor->flat<int64>();
249 auto row_seq_tensor_flat = row_seq_tensor.flat<int64>();
250 auto col_seq_tensor_flat = col_seq_tensor.flat<int64>();
251 auto orig_input_tensor_shape_flat = orig_input_tensor_shape.flat<int64>();
270 in_backprop_tensor_temp.flat<double>().setZero();
273 in_backprop_tensor_temp.flat<double>().data(), in_depth,
275 ConstEigenMatrixMap out_backprop_mat(out_backprop.flat<T>().data()
[all...] |
| pooling_ops_common.cc | 211 auto input_data = AsDeviceMemory(transformed_input.template flat<T>().data(),
212 transformed_input.template flat<T>().size());
214 AsDeviceMemory(transformed_output.template flat<T>().data(),
215 transformed_output.template flat<T>().size());
355 AsDeviceMemory(transformed_output.template flat<T>().data(),
356 transformed_output.template flat<T>().size());
358 AsDeviceMemory(transformed_input.template flat<T>().data(),
359 transformed_input.template flat<T>().size());
361 AsDeviceMemory(transformed_output_backprop.template flat<T>().data(),
362 transformed_output_backprop.template flat<T>().size())
[all...] |
| as_string_op.cc | 116 auto output_flat = output_tensor->flat<string>();
120 const auto& input_flat = input_tensor->flat<T>(); \
133 const auto& input_flat = input_tensor->flat<bool>();
139 const auto& input_flat = input_tensor->flat<complex64>();
|
| data_format_ops.cc | 57 input.flat<T>(), output->flat<T>());
108 context->eigen_device<Device>(), input.flat<T>(), output->flat<T>(),
139 typename TTypes<T>::Flat y); \
|
| dense_update_ops.cc | 40 copy(context->eigen_device<Device>(), lhs->flat<T>(), rhs.flat<T>());
81 update_functor(context->template eigen_device<Device>(), Tparams.flat<T>(),
82 Tupdate.flat<T>());
|
| histogram_op.cc | 47 auto index_to_bin = index_to_bin_tensor.flat<int32>();
92 const auto values = values_tensor.flat<T>();
93 const auto value_range = value_range_tensor.flat<T>();
109 auto out = out_tensor->flat<Tout>();
|
| l2loss_op.cc | 46 (input.flat<T>().square() * static_cast<T>(0.5)).sum();
|
| restore_op_test.cc | 200 EXPECT_EQ(i != 0, output->flat<bool>()(i));
212 EXPECT_EQ(i + 1, output->flat<int32>()(i));
224 EXPECT_EQ(static_cast<float>(i) / 10, output->flat<float>()(i));
236 EXPECT_EQ(static_cast<double>(i) / 20, output->flat<double>()(i));
248 EXPECT_EQ(*reinterpret_cast<qint8*>(&i), output->flat<qint8>()(i));
261 output->flat<qint32>()(i));
273 EXPECT_EQ(i + 1, output->flat<uint8>()(i));
285 EXPECT_EQ(i - 7, output->flat<int8>()(i));
297 EXPECT_EQ(i - 8, output->flat<int16>()(i));
309 EXPECT_EQ(i - 9, output->flat<int64>()(i))
[all...] |
| bias_op.cc | 242 output->template flat<T>().setZero();
258 output->template flat<T>().device(context->eigen_device<Device>()) =
259 output_backprop.flat<T>()
271 output->template flat<T>().device(context->eigen_device<Device>()) =
272 output_backprop.flat<T>()
344 input.flat<T>().data(), bias.flat<T>().data(),
345 output->flat<T>().data(), batch, width, height,
397 output->flat<T>().data(), output->NumElements() * sizeof(T));
401 output_backprop.template flat<T>().data()
[all...] |
| bucketize_op.cc | 64 const auto input = input_tensor.flat<T>();
69 auto output = output_tensor->template flat<int32>();
|
| inplace_ops_functor_gpu.cu.cc | 50 const T* src = value.flat<T>().data();
51 T* dst = output->flat<T>().data();
|
| /external/tensorflow/tensorflow/core/util/ |
| util.cc | 99 string SliceDebugString(const TensorShape& shape, const int64 flat) {
103 if (dims == 1) return strings::StrCat("[", flat, "]");
113 int64 left = flat;
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_test.cc | 132 t2.flat<T>() = t.flat<T>();
158 gtl::ArraySlice<T> values(t.flat<T>().data(), t.NumElements());
219 t.flat<ResourceHandle>()(0) = tmp;
226 value.flat<float>()(0) = 42.0f;
227 t.flat<Variant>()(0) = value;
247 t2.flat<Variant>() = t.flat<Variant>();
252 gtl::ArraySlice<Variant> values(t.flat<Variant>().data(), t.NumElements());
280 internal.flat<float>()(0) = 42.0f
488 auto flat = t.flat<float>(); local
[all...] |
| /external/libvpx/libvpx/vpx_dsp/arm/ |
| highbd_loopfilter_neon.c | 29 // Here flat is 128-bit long, with each 16-bit chunk being a mask of
31 // all 0s or all 1s. We pairwise add flat to a 32-bit long number flat_status.
32 // flat equals 0 if and only if flat_status equals 0.
33 // flat equals -1 (all 1s) if and only if flat_status equals -4. (This is true
35 static INLINE uint32_t calc_flat_status(const uint16x8_t flat) {
36 const uint64x1_t t0 = vadd_u64(vreinterpret_u64_u16(vget_low_u16(flat)),
37 vreinterpret_u64_u16(vget_high_u16(flat)));
73 const uint16x8_t q2, const uint16x8_t q3, uint16x8_t *flat,
78 *flat = vmaxq_u16(max, vabdq_u16(p2, p0));
79 *flat = vmaxq_u16(*flat, vabdq_u16(q2, q0))
616 op2, op1, op0, oq0, oq1, oq2, mask, flat, hev; local
640 op2, op1, op0, oq0, oq1, oq2, mask, flat, hev; local
669 uint16x8_t mask, flat, flat2, hev, p7, p6, p5, p4, p3, p2, p1, p0, q0, q1, q2, local
692 uint16x8_t mask, flat, flat2, hev, p7, p6, p5, p4, p3, p2, p1, p0, q0, q1, q2, local
[all...] |
| /bionic/libc/kernel/uapi/linux/ |
| virtio_input.h | 35 __u32 flat; member in struct:virtio_input_absinfo
|
| /device/linaro/bootloader/edk2/UefiCpuPkg/ResetVector/Vtf0/Ia32/ |
| Flat32ToFlat64.asm | 3 ; Transition from 32 bit flat protected mode into 64 bit flat protected mode
|
| /external/kernel-headers/original/uapi/linux/ |
| virtio_input.h | 46 __u32 flat; member in struct:virtio_input_absinfo
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| brw_meta_util.h | 63 bool flat);
|
| /external/swiftshader/src/Renderer/ |
| QuadRasterizer.hpp | 51 Float4 interpolate(Float4 &x, Float4 &D, Float4 &rhw, Pointer<Byte> planeEquation, bool flat, bool perspective, bool clamp);
|
| /external/tensorflow/tensorflow/contrib/input_pipeline/kernels/ |
| input_pipeline_kernels.cc | 33 auto list_flat = list->flat<string>();
|
| /frameworks/av/media/libstagefright/codecs/mp3dec/include/ |
| pvmp3decoder_api.h | 81 flat = 0, enumerator in enum:__anon45388
|
| /device/linaro/bootloader/edk2/EdkCompatibilityPkg/Foundation/Library/EfiCommonLib/Ia32/ |
| MultU64x32.S | 24 #.MODEL flat,C
|
| /external/libvpx/libvpx/vpx_dsp/mips/ |
| loopfilter_4_msa.c | 19 v16u8 mask, hev, flat, thresh, b_limit, limit; local
30 mask, flat);
47 v16u8 mask, hev, flat, thresh0, b_limit0, limit0, thresh1, b_limit1, limit1; local
66 mask, flat);
76 v16u8 mask, hev, flat, limit, thresh, b_limit; local
89 mask, flat);
107 v16u8 mask, hev, flat; local
135 mask, flat);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| tensor_utils.cc | 93 return sparse_float_feature_shapes_list[0].flat<int64>()(0);
96 return sparse_int_feature_shapes_list[0].flat<int64>()(0);
|
