Home
last modified time | relevance | path

Searched defs:dim_count (Results 1 – 7 of 7) sorted by relevance

/external/OpenCL-CTS/test_conformance/basic/
Dtest_rw_image_access_qualifier.cpp141 int dim_count = 2; in test_rw_image_access_qualifier() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc25 int dim_count = pad_values.size(); in PadAttributeArray() local
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc93 int dim_count = 0; in BitcastOp_Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h37 int dim_count) { in StridedSlicePadIndices()
/external/mesa3d/src/gallium/drivers/virgl/
Dvirgl_transfer_queue.c111 const int dim_count = transfer_dim(xfer); in transfer_overlap() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc244 const int64_t dim_count = ctrl_.count_of_dimensions_to_convert; in GetFirstChosenSpatialDim() local
252 const int64_t dim_count = ctrl_.count_of_dimensions_to_convert; in GetChosenSpatialDims() local
2679 int dim_count = 0; in PropagateOnConv() local
2767 int dim_count = 0; in PropagateOnReverse() local
3754 int dim_count = 0; in PerformSpaceToBatchOnConvolution() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Deinsum_spmd_expander.cc362 absl::flat_hash_map<std::string, int> dim_count = in ComputeLayoutBackward() local