Home
last modified time | relevance | path

Searched +defs:val +defs:axes (Results 1 – 25 of 674) sorted by relevance

12345678910>>...27

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_fold.cpp132 std::vector<int64_t> axes; in runTorchSlice_opset10() local
315 std::vector<int64_t> axes; in runTorchBackendForOnnx() local
364 std::vector<int64_t> axes; in runTorchBackendForOnnx() local
454 std::vector<int64_t> axes; in runTorchBackendForOnnx() local
545 bool isConstant(Value* val, const ValueToParamPairMap& valsToParamsMap) { in isConstant()
571 for (auto val : node->inputs()) { in getValues() local
601 for (auto val : node->inputs()) { in getOnnxConstParentsToRemove() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dmisc.cpp164 auto val = intValue(dim.node()); in getTensorInfo() local
258 const std::vector<ExprHandle>& axes) { in tensorOrConstant()
272 ExprHandle broadcast(const BufHandle& b, const std::vector<ExprHandle>& axes) { in broadcast()
327 [inputs](const std::vector<VarHandle>& axes) { in computeChunk()
362 "aten_transpose", outputShape, [&](const std::vector<VarHandle>& axes) { in computeTranspose()
373 "aten_transpose", outputShape, [&](std::vector<VarHandle> axes) { in computeTranspose()
387 "aten_expand", outputShape, [&](const std::vector<VarHandle>& axes) { in computeExpand()
402 "aten_view", outputShape, [&](const std::vector<VarHandle>& axes) { in computeReshape()
408 "aten_reshape", outputShape, [&](const std::vector<VarHandle>& axes) { in computeReshape()
635 [&](const std::vector<VarHandle>& axes) { in computeCat()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_array_ops.py146 def _array_internal(val, dtype=None, copy=True, ndmin=0): # pylint: disable=redefined-outer-name argument
196 def array(val, dtype=None, copy=True, ndmin=0): # pylint: disable=redefined-outer-name argument
458 def imag(val): argument
684 def real(val): argument
787 def transpose(a, axes=None): argument
797 def adjust_axes(axes, rank): argument
1281 def rot90(m, k=1, axes=(0, 1)): # pylint: disable=missing-docstring argument
H A Dnp_utils.py41 def _canonicalize_axes(axes, rank): argument
83 def isscalar(val): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp400 auto val = toIValue(v).value(); in constant() local
448 auto val = toIValue(v).value(); in toArg() local
713 if (auto val = to<LongImm>(tc.node())) { in tripCount() local
1193 [&](const std::vector<VarHandle>& axes) { in bindInput()
1283 std::vector<ExprHandle> axes(axes_input.begin(), axes_input.end()); in convertSymbolicOutputToCorrectStrides() local
1396 std::vector<ExprHandle> axes(axes_input.begin(), axes_input.end()); in convertStaticShapeOutputToCorrectStrides() local
1421 auto val = toIValue(v).value(); in bindConstant() local
1593 auto _prefer_static_mem = [](const torch::jit::Value* val) { in deduceMemoryLayoutPolicy()
H A Dlowerings.cpp505 auto val = promoteToDtype(value, input.dtype().scalar_type()); in nnc_lowerings_lazy_registration() local
1020 double val = in nnc_lowerings_lazy_registration() local
1106 "aten_remainder", shape, [&](const std::vector<VarHandle>& axes) { in nnc_lowerings_lazy_registration()
1732 [&](const std::vector<VarHandle>& axes) { in nnc_lowerings_lazy_registration()
1786 [&](const std::vector<VarHandle>& axes) { in nnc_lowerings_lazy_registration()
1800 [&](const std::vector<VarHandle>& axes) { in nnc_lowerings_lazy_registration()
/aosp_15_r20/out/soong/.intermediates/prebuilts/sdk/current/androidx/m2repository/androidx/swiperefreshlayout/swiperefreshlayout/1.2.0-alpha01/androidx.swiperefreshlayout_swiperefreshlayout/android_common_apex30/aar/
Dclasses.jar ... synthetic androidx.swiperefreshlayout.widget.CircularProgressDrawable$Ring val$ring final synthetic androidx.swiperefreshlayout.widget. ...
Dandroidx.swiperefreshlayout_swiperefreshlayout.jar ... synthetic androidx.swiperefreshlayout.widget.CircularProgressDrawable$Ring val$ring final synthetic androidx.swiperefreshlayout.widget. ...
/aosp_15_r20/out/soong/.intermediates/prebuilts/sdk/current/androidx/m2repository/androidx/swiperefreshlayout/swiperefreshlayout/1.2.0-alpha01/androidx.swiperefreshlayout_swiperefreshlayout/android_common/aar/
Dclasses.jar ... synthetic androidx.swiperefreshlayout.widget.CircularProgressDrawable$Ring val$ring final synthetic androidx.swiperefreshlayout.widget. ...
Dandroidx.swiperefreshlayout_swiperefreshlayout.jar ... synthetic androidx.swiperefreshlayout.widget.CircularProgressDrawable$Ring val$ring final synthetic androidx.swiperefreshlayout.widget. ...
/aosp_15_r20/external/libtracefs/src/
H A Dtracefs-hist.c286 struct tracefs_hist_axis *axes, in hist_alloc_nd()
343 struct tracefs_hist_axis *axes) in tracefs_hist_alloc_nd()
366 struct tracefs_hist_axis_cnt *axes) in tracefs_hist_alloc_nd_cnt()
689 const char *val) in tracefs_hist_append_filter()
1641 const char *val) in tracefs_synth_append_start_filter()
1664 const char *val) in tracefs_synth_append_end_filter()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_unary.cc38 void ReduceGeneric(bool keep_dims, const std::vector<int>& axes, in ReduceGeneric()
302 const float val = (*input_float_data)[i]; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc167 bool CanFuseConvOrDepthwiseConv(Value filter, Attribute val, in CanFuseConvOrDepthwiseConv()
179 bool CanFuseConvOrDepthwiseConv(Attribute filter, Attribute val, in CanFuseConvOrDepthwiseConv()
327 const mlir::Attribute &axes, in ShapeMatchesReduceWithKeepAxes()
358 const mlir::Attribute &axes) { in AreInputDimensionsOneInAxes()
459 bool IsLastDimensionEqualOne(Value val) { in IsLastDimensionEqualOne()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dtestlib.cc116 Node* Assign(Graph* g, Node* var, Node* val) { in Assign()
126 Node* Cumsum(Graph* g, Node* data, Node* axes, bool exclusive, bool reverse) { in Cumsum()
137 Node* Reduce(Graph* g, const string& reduce, Node* data, Node* axes, in Reduce()
/aosp_15_r20/out/soong/.intermediates/external/apache-xml/apache-xml/android_common_apex31/withres/
Dapache-xml.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/apache/ org/ ...
/aosp_15_r20/out/target/common/obj/JAVA_LIBRARIES/apache-xml.com.android.art.debug_intermediates/
Dclasses.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/apache/ org/ ...
/aosp_15_r20/out/soong/.intermediates/external/apache-xml/apache-xml/android_common_apex31/javac/
Dapache-xml.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/apache/ org/ ...
/aosp_15_r20/prebuilts/module_sdk/art/1/java/
Dapache-xml.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/apache/ org/ ...
/aosp_15_r20/out/soong/.intermediates/prebuilts/sdk/current/support/android-support-swiperefreshlayout-nodeps/android_common/aar/
Dandroid-support-swiperefreshlayout-nodeps.jar ... .support.v4.widget.CircularProgressDrawable$Ring val$ring final synthetic android.support.v4. ...
Dclasses.jar ... .support.v4.widget.CircularProgressDrawable$Ring val$ring final synthetic android.support.v4. ...
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc2269 for (auto val : llvm::enumerate(begin)) { in convertStridedSliceOp() local
2274 for (const auto& val : llvm::enumerate(end)) { in convertStridedSliceOp() local
2609 ArrayRef<int64_t> axes) { in convertGenericReduceOp()
2674 llvm::SmallVector<int64_t> axes; in convertReduceOpCommon() local
2685 Value val = input_value; in convertReduceOpCommon() local
2928 auto val = convertReduceOpCommon<tosa::ReduceSumOp>( in convertReduceMeanOp() local
/aosp_15_r20/out/soong/.intermediates/prebuilts/sdk/current/support/android-support-compat-nodeps/android_common/aar/
Dclasses.jarandroid/support/v4/app/INotificationSideChannel$Stub$Proxy.class INotificationSideChannel.java package ...
Dandroid-support-compat-nodeps.jarandroid/support/v4/app/INotificationSideChannel$Stub$Proxy.class INotificationSideChannel.java package ...
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops.cpp385 int64_t val = static_cast<int64_t>(std::stoll(s->string(), &sz)); in __anonbfe5918f1602() local
2645 c10::List<int64_t> axes = pop(stack).toIntList(); in __anonbfe5918f9102() local
2829 auto val = pop(stack); in __anonbfe5918f9b02() local
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py1453 def mean_reduce_helper(v, axes=axis): argument
2641 def _local_results(self, val): argument

12345678910>>...27