HomeSort by relevance Sort by last modified time
    Searched refs:flat (Results 1 - 25 of 1024) sorted by null

1 2 3 4 5 6 7 8 91011>>

  /external/tensorflow/tensorflow/core/kernels/
quantized_reshape_op_test.cc 55 input.flat<quint8>()(i) = quint8(i);
56 expected.flat<quint8>()(i) = quint8(i);
58 AddInputFromArray<quint8>(input.shape(), input.flat<quint8>());
64 EXPECT_EQ(-10, GetOutput(1)->flat<float>()(0));
65 EXPECT_EQ(20, GetOutput(2)->flat<float>()(0));
requantize.cc 42 const float input_min_float = ctx->input(1).flat<float>()(0);
43 const float input_max_float = ctx->input(2).flat<float>()(0);
44 const float requested_output_min_float = ctx->input(3).flat<float>()(0);
45 const float requested_output_max_float = ctx->input(4).flat<float>()(0);
64 auto input_array = input.flat<T1>();
68 auto output_array = output->flat<T2>();
79 auto input_i32_array = input.flat<qint32>();
83 output->flat<quint8>().data());
92 output_min->flat<float>().setConstant(requested_output_min_float);
93 output_max->flat<float>().setConstant(requested_output_max_float)
    [all...]
encode_wav_op_test.cc 64 const int sample_rate = outputs[1].flat<int32>()(0);
69 EXPECT_NEAR(0.0f, audio.flat<float>()(0), 1e-4f);
70 EXPECT_NEAR(0.5f, audio.flat<float>()(1), 1e-4f);
71 EXPECT_NEAR(1.0f, audio.flat<float>()(2), 1e-4f);
72 EXPECT_NEAR(-1.0f, audio.flat<float>()(3), 1e-4f);
73 EXPECT_NEAR(0.25f, audio.flat<float>()(4), 1e-4f);
74 EXPECT_NEAR(0.75f, audio.flat<float>()(5), 1e-4f);
75 EXPECT_NEAR(1.0f, audio.flat<float>()(6), 1e-4f);
76 EXPECT_NEAR(-0.5f, audio.flat<float>()(7), 1e-4f);
quantized_activation_ops.cc 35 const float min_input = context->input(1).flat<float>()(0);
36 const float max_input = context->input(2).flat<float>()(0);
43 auto input_ui8_array = input.flat<quint8>();
45 min_as_quantized, 255, output->flat<quint8>().data());
47 output->flat<T>().device(context->eigen_cpu_device()) =
48 input.flat<T>().cwiseMax(min_as_quantized).template cast<T>();
53 output_min->flat<float>()(0) = min_input;
56 output_max->flat<float>()(0) = max_input;
68 const float min_input = context->input(1).flat<float>()(0);
69 const float max_input = context->input(2).flat<float>()(0)
    [all...]
quantized_reshape_op.cc 34 const float input_min_float = ctx->input(2).flat<float>()(0);
35 const float input_max_float = ctx->input(3).flat<float>()(0);
38 output_min->flat<float>()(0) = input_min_float;
42 output_max->flat<float>()(0) = input_max_float;
quantized_bias_add_op.cc 41 const float input_min = context->input(2).flat<float>()(0);
42 const float input_max = context->input(3).flat<float>()(0);
43 const float bias_min = context->input(4).flat<float>()(0);
44 const float bias_max = context->input(5).flat<float>()(0);
69 auto input_ui8_array = input.flat<quint8>();
70 auto bias_ui8_array = bias.flat<quint8>();
77 output->flat<qint32>().data());
86 output_min->flat<float>()(0) = total_min;
90 output_max->flat<float>()(0) = total_max;
relu_op.h 40 functor(context->eigen_device<Device>(), input.flat<T>(),
41 output->flat<T>());
88 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(),
89 output->flat<T>());
99 functor(context->eigen_device<Device>(), input.flat<T>(),
100 output->flat<T>());
130 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(),
131 output->flat<T>())
    [all...]
requantization_range_op.cc 36 auto input_array = input.flat<qint32>();
49 const float input_min_float = ctx->input(1).flat<float>()(0);
50 const float input_max_float = ctx->input(2).flat<float>()(0);
68 output_min->flat<float>().setConstant(used_min_float);
69 output_max->flat<float>().setConstant(used_max_float);
slice_op_test.cc 48 begin.flat<int32>()(0) = 10;
49 begin.flat<int32>()(1) = 10;
52 sizes.flat<int32>()(0) = kDim;
53 sizes.flat<int32>()(1) = size;
56 input.flat<T>().setRandom();
quantized_batch_norm_op.cc 39 auto input_flat = input.flat<T1>();
40 auto mean_flat = mean.flat<T1>();
41 auto var_flat = var.flat<T1>();
42 auto beta_flat = beta.flat<T1>();
43 auto gamma_flat = gamma.flat<T1>();
44 auto output_flat = output->flat<T2>();
102 auto input_flat = input.flat<T1>();
103 auto mean_flat = mean.flat<T1>();
104 auto var_flat = var.flat<T1>();
105 auto beta_flat = beta.flat<T1>()
    [all...]
quantized_pooling_ops.cc 63 const float min_input = context->input(1).flat<float>()(0);
64 const float max_input = context->input(2).flat<float>()(0);
84 int32_input.flat<int32>() = tensor_in.flat<T>().template cast<int32>();
89 output->flat<T>() = int32_output.flat<int32>()
96 output_min->flat<float>()(0) = min_input;
99 output_max->flat<float>()(0) = max_input;
115 const float min_input = context->input(1).flat<float>()(0);
116 const float max_input = context->input(2).flat<float>()(0)
    [all...]
quantize_down_and_shrink_range.cc 43 const float input_min_float = ctx->input(1).flat<float>()(0);
44 const float input_max_float = ctx->input(2).flat<float>()(0);
53 auto input_array = input.flat<T1>();
75 auto output_array = output->flat<T2>();
85 auto input_i32_array = input.flat<qint32>();
88 actual_max_float, output->flat<quint8>().data());
96 output_min->flat<float>().setConstant(actual_min_float);
97 output_max->flat<float>().setConstant(actual_max_float);
base64_ops.cc 39 auto input = input_tensor.flat<string>();
40 auto output = output_tensor->flat<string>();
64 auto input = input_tensor.flat<string>();
65 auto output = output_tensor->flat<string>();
cast_op_impl_bfloat.cc 32 BFloat16ToFloat(inp.flat<bfloat16>().data() + start,
33 out->flat<float>().data() + start, end - start);
47 func(ctx->eigen_device<GPUDevice>(), out->flat<float>(),
48 inp.flat<bfloat16>());
decode_csv_op.cc 57 auto records_t = records->flat<string>();
90 output[f]->flat<int32>()(i) = record_defaults[f].flat<int32>()(0);
97 output[f]->flat<int32>()(i) = value;
110 output[f]->flat<int64>()(i) = record_defaults[f].flat<int64>()(0);
117 output[f]->flat<int64>()(i) = value;
129 output[f]->flat<float>()(i) = record_defaults[f].flat<float>()(0);
136 output[f]->flat<float>()(i) = value
    [all...]
  /external/tensorflow/tensorflow/core/framework/
tensor_testutil.h 37 // Constructs a flat tensor with 'vals'.
41 std::copy_n(vals.data(), vals.size(), ret.flat<T>().data());
58 auto flat = tensor->flat<T>(); local
59 CHECK_EQ(flat.size(), vals.size());
60 if (flat.size() > 0) {
61 std::copy_n(vals.data(), vals.size(), flat.data());
68 auto flat = tensor->flat<T>(); local
69 CHECK_EQ(flat.size(), vals.size())
83 auto flat = tensor->flat<T>(); local
92 auto flat = tensor->flat<T>(); local
    [all...]
tensor_util_test.cc 74 x.flat<float>()(0) = 10.0;
78 y.flat<float>()(0) = 20.0;
81 EXPECT_EQ(10.0, x.flat<float>()(0));
84 x.flat<float>()(0) = 30.0;
87 EXPECT_EQ(20.0, y.flat<float>()(0));
92 y.flat<float>()(0) = 40.0;
95 EXPECT_EQ(20.0, z.flat<float>()(0));
96 EXPECT_EQ(30.0, x.flat<float>()(0));
97 EXPECT_EQ(40.0, y.flat<float>()(0));
110 str1.flat<string>()(0) = "foo1"
    [all...]
  /external/tensorflow/tensorflow/contrib/image/kernels/
adjust_hsv_in_yiq_op_gpu.cu.cc 59 delta_h, scale_s, scale_v, tranformation_matrix.flat<float>().data(),
60 tranformation_matrix.flat<float>().size());
64 AsDeviceMemory(input->flat<float>().data(), input->flat<float>().size());
65 auto b_ptr = AsDeviceMemory(tranformation_matrix.flat<float>().data(),
66 tranformation_matrix.flat<float>().size());
67 auto c_ptr = AsDeviceMemory(output->flat<float>().data(),
68 output->flat<float>().size());
  /device/linaro/bootloader/edk2/EdkCompatibilityPkg/Foundation/Library/RuntimeDxe/EfiRuntimeLib/Ia32/
CpuFlushCache.S 25 #.MODEL flat,C
  /toolchain/binutils/binutils-2.27/gas/testsuite/gas/m68hc11/
abi-m68hc11-16-32.d 7 private flags = 0:\[abi=16-bit int, 32-bit double, cpu=HC11\] \[memory=flat\]
abi-m68hc11-16-64.d 7 private flags = 2:\[abi=16-bit int, 64-bit double, cpu=HC11\] \[memory=flat\]
abi-m68hc11-32-32.d 7 private flags = 1:\[abi=32-bit int, 32-bit double, cpu=HC11\] \[memory=flat\]
abi-m68hc11-32-64.d 7 private flags = 3:\[abi=32-bit int, 64-bit double, cpu=HC11\] \[memory=flat\]
  /external/libcap/libcap/
cap_file.c 78 result->u[i].flat[CAP_INHERITABLE]
80 result->u[i].flat[CAP_PERMITTED]
83 result->u[i].flat[CAP_EFFECTIVE]
84 = result->u[i].flat[CAP_INHERITABLE]
85 | result->u[i].flat[CAP_PERMITTED];
89 result->u[i].flat[CAP_INHERITABLE]
90 = result->u[i].flat[CAP_PERMITTED]
91 = result->u[i].flat[CAP_EFFECTIVE] = 0;
142 eff_not_zero |= cap_d->u[i].flat[CAP_EFFECTIVE];
145 if ((cap_d->u[i].flat[CAP_EFFECTIVE
    [all...]
  /frameworks/native/opengl/libs/EGL/
BlobCache_test.cpp 284 uint8_t* flat = new uint8_t[size]; local
285 ASSERT_EQ(OK, mBC->flatten(flat, size));
286 ASSERT_EQ(OK, mBC2->unflatten(flat, size));
287 delete[] flat;
332 uint8_t* flat = new uint8_t[size]; local
333 ASSERT_EQ(OK, mBC->flatten(flat, size));
334 delete[] flat;
354 uint8_t* flat = new uint8_t[size]; local
355 // ASSERT_EQ(BAD_VALUE, mBC->flatten(flat, size));
358 delete[] flat;
366 uint8_t* flat = new uint8_t[size]; local
383 uint8_t* flat = new uint8_t[size]; local
402 uint8_t* flat = new uint8_t[size]; local
421 uint8_t* flat = new uint8_t[size]; local
    [all...]

Completed in 436 milliseconds

1 2 3 4 5 6 7 8 91011>>