Home
last modified time | relevance | path

Searched defs:dim (Results 1 – 25 of 730) sorted by relevance

12345678910>>...30

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc65 /* static */ std::string ToString(const WindowDimension& dim) { in ToString()
109 [](const WindowDimension& dim) { return StrCat(dim.size()); }); in ToString()
113 [](const WindowDimension& dim) { return StrCat(dim.stride()); }); in ToString()
116 add_field(" pad", [](const WindowDimension& dim) { in ToString()
121 add_field(" lhs_dilate", [](const WindowDimension& dim) { in ToString()
126 add_field(" rhs_dilate", [](const WindowDimension& dim) { in ToString()
131 add_field(" rhs_reversal", [](const WindowDimension& dim) { in ToString()
139 for (const auto& dim : window.dimensions()) { in HasStride() local
148 for (const auto& dim : window.dimensions()) { in HasPadding() local
157 return absl::c_all_of(window.dimensions(), [](const WindowDimension& dim) { in HasSymmetricPadding()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_sparse_kernel.cc89 for (int dim = rt_nested_splits_len - 2; dim >= 0; --dim) { in Compute() local
96 for (int dim = 0; dim < index_prefix.size(); ++dim) { in Compute() local
108 int dim = 0; in Compute() local
141 for (int dim = 0; dim < rt_nested_splits_len; ++dim) { in Compute() local
149 for (int dim = 1; dim < rt_dense_values_in.dims(); ++dim) { in Compute() local
204 for (int dim = 1; dim < values_shape.dims(); ++dim) { in MakeIndexSuffixes() local
221 const std::vector<int64_t>& pos, int dim, in IsCompleted()
Dunique_op_test.cc40 TensorProto GetRandomInt32TensorProto(int dim, int max_int) { in GetRandomInt32TensorProto()
52 TensorProto GetRandomInt32TensorProtoWithRepeat(int dim, int repeat, in GetRandomInt32TensorProtoWithRepeat()
68 const int dim = state.range(0); in BM_Unique_INT32() local
91 const int dim = state.range(0); in BM_Unique_INT32_Repeat() local
114 TensorProto GetRandomStringsTensorProto(int dim, int max_str_len) { in GetRandomStringsTensorProto()
132 const int dim = state.range(0); in BM_Unique_STRING() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes.cc36 bool IsKnown(const TensorShapeProto::Dim& dim) { return dim.size() >= 0; } in IsKnown()
38 bool IsKnownSymbolically(const TensorShapeProto::Dim& dim) { in IsKnownSymbolically()
42 bool IsUnknown(const TensorShapeProto::Dim& dim) { return dim.size() == -1; } in IsUnknown()
48 [](const TensorShapeProto::Dim& dim) { return !IsUnknown(dim); }); in ShapeIsSymbolicallyDefined()
67 for (const auto& dim : shape.dim()) { in NumCoefficients() local
125 for (const auto& dim : bcast.output_shape()) { in ShapeAfterBroadcast() local
157 const auto& dim = shape.dim(i); in CompareSymbolicallyShapedTensorSizes() local
226 for (const auto& dim : numerator.dim()) { in ComputeSizeRatio() local
236 for (const auto& dim : denominator.dim()) { in ComputeSizeRatio() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsharding_util_ops.cc80 for (int dim = 0; dim < expected_rank; ++dim) { in GetAndValidateAttributes() local
108 auto divisor = [&](const int dim) { in GetSliceIndices()
116 for (int dim = num_partitions.size() - 1; dim > 0; --dim) { in GetSliceIndices() local
153 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
166 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
185 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
199 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
208 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
349 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
373 for (int dim = 0; dim < rank; ++dim) { in CompileInternal() local
[all …]
Dreshape_op.cc86 for (int dim = 0; dim < input_shape.dims(); dim++) { in Compile() local
164 for (int64_t dim = start.first; dim < end.first; ++dim) { in Compile() local
174 for (int64_t dim = start.second; dim < end.second; ++dim) { in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc66 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput() local
101 int64_t dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput() local
138 int64_t dim = conv_dnums.kernel_spatial_dimensions(i); in MaybePaddedKernel() local
171 WindowDimension* dim = new_conv_window.mutable_dimensions(i); in CanonicalizeForwardConvolution() local
239 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
323 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution() local
368 int64_t dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution() local
Dgpu_conv_runner.cc370 for (const WindowDimension& dim : window.dimensions()) { in GetGpuConvConfig() local
400 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvConfig() local
415 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvConfig() local
424 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvConfig() local
444 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvConfig() local
451 for (int dim = 0; dim < effective_num_dimensions - num_dimensions; dim++) { in GetGpuConvConfig() local
/external/libaom/av1/encoder/
Dpalette.h29 #define AV1_K_MEANS_RENAME(func, dim) func##_dim##dim##_c argument
56 int n, int k, int dim) { in av1_calc_indices()
89 uint8_t *indices, int n, int k, int dim, in av1_k_means()
/external/mesa3d/src/gallium/auxiliary/tgsi/
Dtgsi_util.c306 unsigned dim_layer_shadow, dim; in tgsi_util_get_inst_usage_mask() local
364 unsigned dim = tgsi_util_get_texture_coord_dim(inst->Memory.Texture); in tgsi_util_get_inst_usage_mask() local
371 unsigned dim = tgsi_util_get_texture_coord_dim(inst->Memory.Texture); in tgsi_util_get_inst_usage_mask() local
392 unsigned dim = tgsi_util_get_texture_coord_dim(inst->Memory.Texture); in tgsi_util_get_inst_usage_mask() local
/external/javassist/src/main/javassist/compiler/ast/
DDeclarator.java33 public Declarator(int type, int dim) { in Declarator()
41 public Declarator(ASTList className, int dim) { in Declarator()
51 public Declarator(int type, String jvmClassName, int dim, in Declarator()
62 public Declarator make(Symbol sym, int dim, ASTree init) { in make()
DInstanceOfExpr.java28 public InstanceOfExpr(ASTList className, int dim, ASTree expr) { in InstanceOfExpr()
32 public InstanceOfExpr(int type, int dim, ASTree expr) { in InstanceOfExpr()
/external/javassist/src/main/javassist/compiler/
DParser.java208 int dim = parseArrayDimension(); in parseFormalType() local
212 int dim = parseArrayDimension(); in parseFormalType() local
591 int dim = parseArrayDimension(); in parseDeclarationOrExpression() local
599 int dim = parseArrayDimension(); in parseDeclarationOrExpression() local
657 int dim = parseArrayDimension(); in parseDeclarator() local
806 int dim = parseArrayDimension(); in parseInstanceOf() local
810 int dim = parseArrayDimension(); in parseInstanceOf() local
917 int dim = parseArrayDimension(); in parseCast() local
926 int dim = parseArrayDimension(); in parseCast() local
1054 int dim = parseArrayDimension(); in parsePostfix() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_normalization.cc99 [&](int dim) { in HandleConcatenate()
115 [&](int dim) { in HandleConcatenate()
129 [&](int dim) { return dim == 1; }); in HandleConcatenate()
392 [&](int dim) { in HandlePad()
412 for (int dim = 0; dim < s_normalized.dimensions_size(); dim++) { in HandlePad() local
416 for (int dim = 0; dim < s.dimensions_size(); dim++) { in HandlePad() local
448 for (int64_t dim : hlo->dimensions()) { in TransformDimensionsForLayoutPreservingHlo() local
464 int DegenDimsUpTo(const Shape& shape, int dim) { in DegenDimsUpTo()
Dconvolution_4d_expander.cc103 for (int64_t dim : removed_input_dimensions) { in ExpandInstruction() local
108 for (int64_t dim : removed_kernel_dimensions) { in ExpandInstruction() local
112 for (int64_t dim : removed_output_dimensions) { in ExpandInstruction() local
Ddot_as_convolution_util.cc142 for (const auto& dim : dot_dnums.batch_dims) { in CreateShardedConvForDotGeneralConvolution() local
149 for (const auto& dim : dot_dnums.contracting_dims) { in CreateShardedConvForDotGeneralConvolution() local
157 for (const auto& dim : dot_dnums.rhs_non_contracting_dims) { in CreateShardedConvForDotGeneralConvolution() local
/external/freetype/src/autofit/
Dafcjk.c84 int dim; in af_cjk_metrics_init_widths() local
664 AF_Dimension dim ) in af_cjk_metrics_scale_dim()
800 AF_Dimension dim ) in af_cjk_hints_compute_segments()
844 AF_Dimension dim ) in af_cjk_hints_link_segments()
1002 AF_Dimension dim ) in af_cjk_hints_compute_edges()
1271 AF_Dimension dim ) in af_cjk_hints_detect_features()
1292 AF_Dimension dim ) in af_cjk_hints_compute_blue_edges()
1497 AF_Dimension dim, in af_cjk_compute_stem_width()
1618 AF_Dimension dim, in af_cjk_align_linked_edge()
1676 AF_Dimension dim ) in af_hint_normal_stem()
[all …]
/external/freetype/src/pshinter/
Dpshrec.c567 ps_dimension_end_mask( PS_Dimension dim, in ps_dimension_end_mask()
586 ps_dimension_reset_mask( PS_Dimension dim, in ps_dimension_reset_mask()
603 ps_dimension_set_mask_bits( PS_Dimension dim, in ps_dimension_set_mask_bits()
629 ps_dimension_add_t1stem( PS_Dimension dim, in ps_dimension_add_t1stem()
697 ps_dimension_add_counter( PS_Dimension dim, in ps_dimension_add_counter()
746 ps_dimension_end( PS_Dimension dim, in ps_dimension_end()
811 PS_Dimension dim; in ps_hints_stem() local
863 PS_Dimension dim; in ps_hints_t1stem3() local
966 PS_Dimension dim = hints->dimension; in ps_hints_t2mask() local
1011 PS_Dimension dim = hints->dimension; in ps_hints_t2counter() local
[all …]
Dpshglob.c44 PSH_Dimension dim = &globals->dimension[direction]; in psh_globals_scale_widths() local
319 FT_Int dim, top, bot, delta; in psh_blues_set_zones() local
663 PSH_Dimension dim = &globals->dimension[1]; in psh_globals_new() local
683 PSH_Dimension dim = &globals->dimension[0]; in psh_globals_new() local
760 PSH_Dimension dim; in psh_globals_set_scale() local
/external/skia/src/codec/
DSkScalingCodec.h20 SkISize dim = this->dimensions(); in onGetScaledDimensions() local
29 SkISize dim = this->dimensions(); in onDimensionsSupported() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dreduce_spmd_expander.cc85 for (const auto& dim : input.sharding_specs()) { in ReducedMeshDimensions() local
88 for (const auto& dim : output.sharding_specs()) { in ReducedMeshDimensions() local
119 for (int dim = 0; dim < rank - 1; ++dim) { in ExtractDims() local
123 for (int dim = 0; dim < rank; ++dim) { in ExtractDims() local
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h222 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex()
236 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice()
529 const int dim = input_tensor.indices().matrix<int64_t>()(i, split_dim); in Split() local
553 const int dim = input_indices_t(i, split_dim); in Split() local
587 for (int dim = 0; dim < dims; dim++) { in Slice() local
622 for (int dim = 0; dim < dims; dim++) { in Slice() local
649 for (int dim = 0; dim < dims; dim++) { in Slice() local
660 for (int dim = 0; dim < dims; dim++) { in Slice() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h311 StorageType::value_type dim(size_t idx) const { return storage_[idx]; } in dim() function
314 StorageType::value_type& dim(size_t idx) { return storage_[idx]; } in dim() function
318 DimsAdapter& Append(int32_t dim) { in Append()
326 DimsAdapter& Prepend(std::optional<int32_t> dim) { in Prepend()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Drestrict_max_rank.cc115 for (int64_t dim = 0; dim < spatialDims; dim++) { in matchAndRewrite() local
135 for (int64_t dim = 0; dim < spatialDims; dim++) { in matchAndRewrite() local
148 for (int dim = spatialDims - 1; dim >= 0; dim--) { in matchAndRewrite() local
/external/javassist/src/main/javassist/bytecode/stackmap/
DTypeData.java88 public abstract TypeData getArrayType(int dim) throws NotFoundException; in getArrayType()
109 protected TypeVar toTypeVar(int dim) { return null; } in toTypeVar()
175 public TypeData getArrayType(int dim) throws NotFoundException { in getArrayType()
305 protected TypeVar toTypeVar(int dim) { in toTypeVar()
314 public TypeData getArrayType(int dim) throws NotFoundException { in getArrayType()
666 protected TypeVar toTypeVar(int dim) { return element.toTypeVar(dim + 1); } in toTypeVar()
669 public TypeData getArrayType(int dim) throws NotFoundException { in getArrayType()
753 protected TypeVar toTypeVar(int dim) { return array.toTypeVar(dim - 1); } in toTypeVar()
756 public TypeData getArrayType(int dim) throws NotFoundException { in getArrayType()
791 protected TypeVar toTypeVar(int dim) { return null; } in toTypeVar()
[all …]

12345678910>>...30