| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| shape_util.cc | 26 const int dims = input_shape.dims(); local
29 for (int i = 0; i < dims; ++i) {
40 for (int i = 0; i < dims; ++i) {
|
| fill_op.cc | 37 const TensorShape dims_shape = ctx->InputShape("dims");
41 errors::InvalidArgument("dims must be a vector of int32, got shape ",
47 std::vector<int64> dims; variable
48 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("dims", &dims));
50 auto result = xla::Broadcast(ctx->Input("value"), dims);
55 REGISTER_XLA_OP(Name("Fill").CompileTimeConstantInput("dims"), FillOp);
|
| pad_op.cc | 35 const int dims = input_shape.dims(); variable
42 ctx, dims == pad_shape.dim_size(0),
48 if (dims == 0) {
59 for (int i = 0; i < dims; ++i) {
|
| mirror_pad_op.cc | 75 const int dims = input_shape.dims(); variable
82 ctx, dims == pad_shape.dim_size(0),
|
| transpose_op.cc | 48 const int dims = input_shape.dims(); variable
49 OP_REQUIRES(ctx, dims == perm_tensor_shape.num_elements(),
51 input_shape.dims(),
59 // Check whether permutation is a permutation of integers of [0 .. dims).
60 absl::InlinedVector<bool, 8> bits(dims);
62 for (int i = 0; i < dims; ++i) {
65 ctx, 0 <= d && d < dims,
66 errors::InvalidArgument(d, " is out of range [0 .. ", dims, ")"));
73 for (int i = 0; i < dims; ++i)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| scan_ops.h | 34 Eigen::array<bool, 3> dims; local
35 dims[0] = false;
36 dims[1] = reverse;
37 dims[2] = false;
39 To32Bit(in).reverse(dims).scan(1, reducer, exclusive).reverse(dims);
|
| reduction_ops_common.cc | 36 const int dims = data_reshape_.size(); local
38 for (int i = reduce_first_axis_; i < dims; i += 2) {
41 for (int i = !reduce_first_axis_; i < dims; i += 2) {
48 const int dims = data_reshape_.size(); local
49 const int unreduced_dims = (dims + !reduce_first_axis_) / 2;
50 gtl::InlinedVector<int32, 8> perm(dims);
54 for (int i = unreduced_dims; i < dims; i++) {
66 if (index < -data.dims() || index >= data.dims()) {
68 " for input with ", data.dims(),
[all...] |
| roll_op_test.cc | 437 const int dims = static_cast<int>(input.dims()); local
438 Tensor shift(DT_INT32, TensorShape({dims}));
439 for (int i = 0; i < dims; i++) {
443 Tensor axis(DT_INT32, TensorShape({dims}));
444 for (int i = 0; i < dims; i++) {
|
| unravel_index_op.cc | 55 auto dims = dims_tensor.vec<Tidx>(); variable
66 strides = dims.reverse(reverse)
77 strides_shifted = dims.reverse(reverse)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| tensor_ctypes.h | 104 TfLiteIntArray* dims = tensor->dims; local
105 const int dims_size = dims->size;
106 const int32_t* dims_data = dims->data;
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_pad_attributes.cc | 45 const std::vector<int>& dims = array.shape().dims(); local
46 CHECK_EQ(dims.size(), 2);
50 for (int i = 0; i < dims[0]; ++i) {
|
| resolve_padv2_attributes.cc | 45 const std::vector<int>& dims = array.shape().dims(); local
46 CHECK_EQ(dims.size(), 2);
50 for (int i = 0; i < dims[0]; ++i) {
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_forced_eval.cpp | 32 Eigen::array<DimPair, 1> dims; local
33 dims[0] = DimPair(1, 0);
35 mat3 = mat3.contract(mat2, dims).eval();
|
| cxx11_tensor_convolution.cpp | 54 Eigen::array<ptrdiff_t, 2> dims; local
55 dims[0] = 0;
56 dims[1] = 1;
57 result = input.convolve(kernel, dims);
80 Eigen::array<ptrdiff_t, 1> dims; local
81 dims[0] = 0;
88 valid = input.pad(padding).convolve(kernel, dims);
96 same = input.pad(padding).convolve(kernel, dims);
106 full = input.pad(padding).convolve(kernel, dims);
122 Eigen::array<ptrdiff_t, 1> dims; local
[all...] |
| cxx11_tensor_custom_op.cpp | 75 array<DimPair, 1> dims; local
76 dims[0] = DimPair(1, 0);
78 output.template chip<2>(i).device(device) = input1.chip<2>(i).contract(input2.chip<2>(i), dims);
94 array<DimPair, 1> dims; local
95 dims[0] = DimPair(1, 0);
96 Tensor<float, 2> reference = tensor1.chip<2>(i).contract(tensor2.chip<2>(i), dims);
|
| cxx11_tensor_of_complex.cpp | 86 Eigen::array<DimPair, 2> dims; local
87 dims[0] = DimPair(2, 0);
88 dims[1] = DimPair(3, 1);
89 t_result = t_left.contract(t_right, dims);
|
| cxx11_tensor_simple.cpp | 198 array<Eigen::DenseIndex, 3> dims; local
199 dims[0] = 2;
200 dims[1] = 3;
201 dims[2] = 4;
202 Tensor<int, 3> t1(dims);
203 Tensor<int, 3, RowMajor> t2(dims);
|
| cxx11_tensor_argmax.cpp | 65 DimensionList<DenseIndex, 4> dims; local
67 dims, internal::ArgMaxTupleReducer<Tuple<DenseIndex, float> >());
97 DimensionList<DenseIndex, 4> dims; local
99 dims, internal::ArgMinTupleReducer<Tuple<DenseIndex, float> >());
164 std::vector<int> dims {2, 3, 5, 7};
222 std::vector<int> dims {2, 3, 5, 7};
|
| /device/generic/goldfish-opengl/shared/OpenglCodecCommon/ |
| TextureSharedData.h | 35 TextureDims* dims; member in struct:TextureRec
|
| /external/libtextclassifier/utils/ |
| tensor-view.h | 50 int dims() const { return shape_.size(); } function in class:libtextclassifier3::TensorView
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_rewriter_transform.cc | 57 std::vector<int64> dims; local
58 CHECK(str_util::SplitAndParseAsInts(shape_string, ',', &dims));
72 inputs.emplace_back(input_name, Tensor(data_type, TensorShape(dims)));
|
| /external/webrtc/webrtc/modules/audio_coding/codecs/isac/main/source/ |
| transform.c | 45 int dims[1]; local
50 dims[0] = FRAMESAMPLES_HALF;
64 WebRtcIsac_Fftns(1, dims, tmpre, tmpim, -1, 1.0, fftstr_obj);
91 int dims; local
93 dims = FRAMESAMPLES_HALF;
113 WebRtcIsac_Fftns(1, &dims, outre1, outre2, 1, FRAMESAMPLES_HALF, fftstr_obj);
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_test_util.cc | 61 int64_t dims[] = {2, 2}; local
64 TF_DOUBLE, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data));
75 int64_t dims[] = {2, 2}; local
78 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data));
89 int64_t dims[] = {3, 2}; local
92 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data));
103 int64_t dims[] = {3, 2} local
145 int64_t dims[] = {1}; local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| partial_tensor_shape_test.cc | 30 EXPECT_EQ(s.dims(), -1);
36 ASSERT_EQ(2, s.dims());
42 ASSERT_EQ(4, s1.dims());
51 ASSERT_EQ(3, s2.dims());
52 ASSERT_EQ(4, s3.dims());
64 EXPECT_EQ(-1, s4.dims());
115 EXPECT_EQ(full.dims(), 0);
117 EXPECT_EQ(full.dims(), 2);
203 EXPECT_EQ(test.dims(), 3);
210 EXPECT_EQ(test.dims(), 3)
249 const int64 dims[1] = {}; local
258 const int64 dims[3] = {7, -1, 2}; local
269 const int64 dims[3] = {7, -2, 2}; local
[all...] |
| tensor_slice.h | 63 int dims() const { return starts_.size(); } function in class:tensorflow::TensorSlice
67 DCHECK_LT(d, dims());
73 DCHECK_LT(d, dims());
79 DCHECK_LT(d, dims());
85 DCHECK_LT(d, dims());
92 DCHECK_LT(d, dims());
108 // Requires: dim >= dims().
118 // We allow NDIMS to be greater than dims(), in which case we will pad the
202 CHECK_EQ(shape.dims(), dims()) << "Incompatible dimensions between shape
[all...] |
