Home
last modified time | relevance | path

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

12345678910>>...17

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc51 /* static */ string ToString(const WindowDimension& dim) { in ToString()
94 [](const WindowDimension& dim) { return StrCat(dim.size()); }); in ToString()
97 [](const WindowDimension& dim) { return StrCat(dim.stride()); }); in ToString()
100 add_field(" pad", [](const WindowDimension& dim) { in ToString()
105 add_field(" lhs_dilate", [](const WindowDimension& dim) { in ToString()
110 add_field(" rhs_dilate", [](const WindowDimension& dim) { in ToString()
115 add_field(" rhs_reversal", [](const WindowDimension& dim) { in ToString()
123 for (const auto& dim : window.dimensions()) { in HasStride() local
132 for (const auto& dim : window.dimensions()) { in HasPadding() local
141 return absl::c_all_of(window.dimensions(), [](const WindowDimension& dim) { in HasSymmetricPadding()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dunique_op_test.cc38 TensorProto GetRandomInt32TensorProto(int dim, int max_int) { in GetRandomInt32TensorProto()
50 TensorProto GetRandomInt32TensorProtoWithRepeat(int dim, int repeat, in GetRandomInt32TensorProtoWithRepeat()
65 static void BM_Unique_INT32(int iters, int dim, int max_int) { in BM_Unique_INT32()
84 static void BM_Unique_INT32_Repeat(int iters, int dim, int max_int) { in BM_Unique_INT32_Repeat()
105 TensorProto GetRandomStringsTensorProto(int dim, int max_str_len) { in GetRandomStringsTensorProto()
122 static void BM_Unique_STRING(int iters, int dim) { in BM_Unique_STRING()
Dragged_tensor_to_sparse_kernel.cc84 for (int dim = rt_nested_splits_len - 2; dim >= 0; --dim) { in Compute() local
91 for (int dim = 0; dim < index_prefix.size(); ++dim) { in Compute() local
103 int dim = 0; in Compute() local
136 for (int dim = 0; dim < rt_nested_splits_len; ++dim) { in Compute() local
144 for (int dim = 1; dim < rt_dense_values_in.dims(); ++dim) { in Compute() local
191 for (int dim = 1; dim < values_shape.dims(); ++dim) { in MakeIndexSuffixes() local
208 const std::vector<int64>& pos, int dim, in IsCompleted()
/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes.cc33 bool IsKnown(const TensorShapeProto::Dim& dim) { return dim.size() >= 0; } in IsKnown()
35 bool IsKnownSymbolically(const TensorShapeProto::Dim& dim) { in IsKnownSymbolically()
39 bool IsUnknown(const TensorShapeProto::Dim& dim) { return dim.size() == -1; } in IsUnknown()
45 [](const TensorShapeProto::Dim& dim) { return !IsUnknown(dim); }); in ShapeIsSymbolicallyDefined()
64 for (const auto& dim : shape.dim()) { in NumCoefficients() local
122 for (const auto& dim : bcast.output_shape()) { in ShapeAfterBroadcast() local
154 const auto& dim = shape.dim(i); in CompareSymbolicallyShapedTensorSizes() local
223 for (const auto& dim : numerator.dim()) { in ComputeSizeRatio() local
233 for (const auto& dim : denominator.dim()) { in ComputeSizeRatio() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dtrt_plugin.cc39 nvinfer1::Dims dim; in PluginTensorRT() local
54 nvinfer1::Dims dim; in configure() local
65 nvinfer1::Dims dim; in getSerializationSize() local
85 auto dim = input_dim_list_[i]; in serialize() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h61 inline int64 GetDim(absl::Span<const int64> data, DimIndex dim) { in GetDim()
65 inline void SetDim(absl::Span<int64> data, DimIndex dim, int64 value) { in SetDim()
69 inline void SetDim(std::vector<int64>* data, DimIndex dim, int64 value) { in SetDim()
/external/libaom/libaom/av1/encoder/
Dpalette.h21 #define AV1_K_MEANS_RENAME(func, dim) func##_dim##dim argument
39 uint8_t *indices, int n, int k, int dim) { in av1_calc_indices()
54 uint8_t *indices, int n, int k, int dim, in av1_k_means()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc65 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput() local
99 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput() local
136 int64 dim = conv_dnums.kernel_spatial_dimensions(i); in MaybePaddedKernel() local
168 WindowDimension* dim = new_conv_window.mutable_dimensions(i); in CanonicalizeForwardConvolution() local
236 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
316 int64 dim = backward_conv_dnums.output_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution() local
356 int64 dim = backward_conv_dnums.output_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution() local
Dcudnn_conv_runner.cc159 for (const WindowDimension& dim : window.dimensions()) { in RunCudnnConvImpl() local
186 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl() local
199 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl() local
208 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl() local
223 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl() local
/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
391 unsigned dim = tgsi_util_get_texture_coord_dim(inst->Memory.Texture); in tgsi_util_get_inst_usage_mask() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.cc22 xla::StatusOr<xla::XlaOp> Contract(xla::XlaOp input, int64 dim) { in Contract()
48 xla::StatusOr<xla::XlaOp> Expand(xla::XlaOp input, int64 dim) { in Expand()
Dscatter.cc62 for (int64 dim : indices_dims) { in XlaScatter() local
152 for (int64 dim = num_index_dims; dim < buffer_rank; ++dim) { in XlaScatter() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc37 SpatialDimensionOutputSizeAndPadding dim; in GetWindowedOutputSize() local
81 SpatialDimensionOutputSizeAndPadding dim; in ConvGradExtractAndVerifyDimension() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.h64 int64 dim; member
84 void SetDynamicSize(HloInstruction* inst, const ShapeIndex& index, int64 dim, in SetDynamicSize()
/external/freetype/src/pshinter/
Dpshrec.c590 ps_dimension_end_mask( PS_Dimension dim, in ps_dimension_end_mask()
609 ps_dimension_reset_mask( PS_Dimension dim, in ps_dimension_reset_mask()
626 ps_dimension_set_mask_bits( PS_Dimension dim, in ps_dimension_set_mask_bits()
652 ps_dimension_add_t1stem( PS_Dimension dim, in ps_dimension_add_t1stem()
723 ps_dimension_add_counter( PS_Dimension dim, in ps_dimension_add_counter()
781 ps_dimension_end( PS_Dimension dim, in ps_dimension_end()
846 PS_Dimension dim; in ps_hints_stem() local
898 PS_Dimension dim; in ps_hints_t1stem3() local
1001 PS_Dimension dim = hints->dimension; in ps_hints_t2mask() local
1046 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/freetype/src/autofit/
Dafcjk.c90 int dim; in af_cjk_metrics_init_widths() local
671 AF_Dimension dim ) in af_cjk_metrics_scale_dim()
805 AF_Dimension dim ) in af_cjk_hints_compute_segments()
849 AF_Dimension dim ) in af_cjk_hints_link_segments()
1007 AF_Dimension dim ) in af_cjk_hints_compute_edges()
1273 AF_Dimension dim ) in af_cjk_hints_detect_features()
1294 AF_Dimension dim ) in af_cjk_hints_compute_blue_edges()
1510 AF_Dimension dim, in af_cjk_compute_stem_width()
1631 AF_Dimension dim, in af_cjk_align_linked_edge()
1687 AF_Dimension dim ) in af_hint_normal_stem()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc38 for (auto dim : s) { in AppendTo() local
145 static inline bool Set16(bool partial, uint16* dst, int dim, int64 val) { in Set16()
277 uint16 dim = as16()->dims_[d]; in dim_size() local
281 uint32 dim = as32()->dims_[d]; in dim_size() local
516 int64 dim = shape.dim_size(i); in DebugString() local
579 T dim = internal::SubtleMustCopy(dims[i]); in MakeShapeHelper() local
646 for (auto dim : shape) out.AddDim(dim.size); in Concatenate() local
758 for (auto dim : shape) { in NumElements() local
Dtensor_slice.cc26 TensorSlice::TensorSlice(int dim) { SetFullSlice(dim); } in TensorSlice()
92 void TensorSlice::SetFullSlice(int dim) { in SetFullSlice()
102 void TensorSlice::Extend(int dim) { in Extend()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc38 for (auto dim : xla::LayoutUtil::MinorToMajor(shape)) { in PopulateInfeedLayoutVector() local
53 [](int64 dim) { return dim == -1; })) { in MakeLayout()
57 for (auto dim : minor_to_major) { in MakeLayout() local
/external/eigen/test/
Dumeyama.cpp91 void run_test(int dim, int num_elements) in run_test()
129 const int dim = Dimension; in run_fixed_size_test() local
165 for (int dim=2; dim<8; ++dim) in test_umeyama() local
Dbug1213_main.cpp13 bool bug1213_2(const Eigen::Matrix<T,dim,1>& ) in bug1213_2() argument
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_tensor.h55 bool AddValue(const T& value, TFProfTensorProto* dim) { in AddValue()
86 TFProfTensorProto* dim) { in BuildOutput()
/external/mesa3d/src/mesa/math/
Dm_eval.c75 GLuint dim, GLuint order) in _math_horner_bezier_curve()
119 GLuint dim, GLuint uorder, GLuint vorder) in _math_horner_bezier_surf()
196 GLfloat * dv, GLfloat u, GLfloat v, GLuint dim, in _math_de_casteljau_surf()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h329 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex()
343 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice()
607 const int dim = input_tensor.indices().matrix<int64>()(i, split_dim); in Split() local
627 const int dim = input_indices_t(i, split_dim); in Split() local
661 for (int dim = 0; dim < dims; dim++) { in Slice() local
682 for (int dim = 0; dim < dims; dim++) { in Slice() local
709 for (int dim = 0; dim < dims; dim++) { in Slice() local
720 for (int dim = 0; dim < dims; dim++) { in Slice() local

12345678910>>...17