Home
last modified time | relevance | path

Searched refs:dim_y (Results 1 – 6 of 6) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction_cuda.cu62 int dim_y = 1; in test_first_dim_reductions() local
65 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_first_dim_reductions()
77 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_first_dim_reductions()
78 Type* out_data = (Type*)dev.allocate(dim_z*dim_y*sizeof(Type)); in test_first_dim_reductions()
79 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions()
80 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_y, dim_z); in test_first_dim_reductions()
86 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions()
102 int dim_y = 1; in test_last_dim_reductions() local
105 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_last_dim_reductions()
117 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_last_dim_reductions()
[all …]
Dcxx11_tensor_reduction_sycl.cpp58 int dim_y = 1; in test_first_dim_reductions_sycl() local
61 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_first_dim_reductions_sycl()
64 array<int, 2> reduced_tensorRange = {{dim_y, dim_z}}; in test_first_dim_reductions_sycl()
96 int dim_y = 1; in test_last_dim_reductions_sycl() local
99 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_last_dim_reductions_sycl()
102 array<int, 2> reduced_tensorRange = {{dim_x, dim_y}}; in test_last_dim_reductions_sycl()
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py522 for (dim_x, dim_y) in broadcasted_dims:
523 if dim_x.value is None or dim_y.value is None:
531 elif dim_y.value is not None and dim_y.value > 1:
532 return_dims.append(dim_y)
537 return_dims.append(dim_y)
538 elif dim_y.value == 1:
541 elif dim_x.value == dim_y.value:
544 return_dims.append(dim_x.merge_with(dim_y))
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1235 const auto dim_y = in BroadcastBinaryOpShapeFn() local
1237 if (!c->ValueKnown(dim_x) || !c->ValueKnown(dim_y)) { in BroadcastBinaryOpShapeFn()
1248 } else if (c->Value(dim_y) > 1) { in BroadcastBinaryOpShapeFn()
1249 dims.push_back(dim_y); in BroadcastBinaryOpShapeFn()
1251 dims.push_back(dim_y); in BroadcastBinaryOpShapeFn()
1252 } else if (c->Value(dim_y) == 1) { in BroadcastBinaryOpShapeFn()
1254 } else if (dim_y.SameHandle(dim_x)) { in BroadcastBinaryOpShapeFn()
1259 } else if (c->Value(dim_x) == 1 || c->Value(dim_y) == 1) { in BroadcastBinaryOpShapeFn()
1262 dims.push_back(dim_y); in BroadcastBinaryOpShapeFn()
1264 DCHECK_EQ(c->Value(dim_y), 1); in BroadcastBinaryOpShapeFn()
[all …]
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc82 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y)); in ComputeBinaryOperatorOutputSize() local
83 if (dim_x == -1 || dim_y == -1) { in ComputeBinaryOperatorOutputSize()
86 } else if (dim_x == 1 || dim_y == 1) { in ComputeBinaryOperatorOutputSize()
90 dims_out->push_back(dim_y); in ComputeBinaryOperatorOutputSize()
93 DCHECK_EQ(dim_y, 1); in ComputeBinaryOperatorOutputSize()
98 CHECK_EQ(dim_x, dim_y) << "Dimensions must match"; in ComputeBinaryOperatorOutputSize()
/external/freetype/src/pshinter/
Dpshalgo.c2114 PSH_Dimension dim_y = &glyph->globals->dimension[1]; in ps_hints_apply() local
2117 FT_Fixed y_scale = dim_y->scale_mult; in ps_hints_apply()