- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 172 for Axis (0.04 sec)
-
tensorflow/compiler/mlir/tfr/tests/control_flow.mlir
%n: i32 {tfr.name="N"}, %axis: i32 {tfr.name="axis"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i32 %eq = arith.cmpi eq, %n, %cst : i32 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor %temp = tfr.call @tf__expand_dims(%v1, %axis) : (!tfr.tensor, i32) -> !tfr.tensor %res = scf.if %eq -> !tfr.tensor {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:58:25 UTC 2022 - 3.2K bytes - Viewed (0) -
testing/internal-performance-testing/src/main/resources/org/gradle/reporting/flot.selection.min.js
rRedrawOverlay();if(!preventEvent)plot.getPlaceholder().trigger("plotunselected",[])}}function extractRange(ranges,coord){var axis,from,to,key,axes=plot.getAxes();for(var k in axes){axis=axes[k];if(axis.direction==coord){key=coord+axis.n+"axis";if(!ranges[key]&&axis.n==1)key=coord+"axis";if(ranges[key]){from=ranges[key].from;to=ranges[key].to;break}}}if(!ranges[key]){axis=coord=="x"?plot.getXAxes()[0]:plot.getYAxes()[0];from=ranges[coord+"1"];to=ranges[coord+"2"]}if(from!=null&&to!=null&&from>to){var...
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Apr 04 07:21:38 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/pad/ops_defs.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 01 05:00:29 UTC 2021 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc
StridedArrayViewBase(ArrayRef<int64_t> shape, ArrayRef<int64_t> index, int64_t axis) { assert(shape.size() == index.size()); assert(axis < shape.size()); assert(axis >= 0); assert(index[axis] == 0); offset_ = IndexToOffset(shape, index); stride_ = StrideForAxis(shape, axis); size_ = shape[axis]; } // Returns the size of the 1-d slice across the tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/cc/gradients/linalg_grad.cc
auto axis = EinsumGetAxisFromLabel(subscripts, s); if (!axis.has_value()) { // Should never happen. scope.UpdateStatus(errors::Internal( absl::StrCat("Missing axis", absl::string_view(&s, 1)))); } else { reduced_axes.push_back(*axis); } } // Get the corresponding dimensions for each reduced axis. std::vector<Output> reduced_dims_inputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 07 23:11:54 UTC 2022 - 20.4K bytes - Viewed (0) -
guava/src/com/google/common/collect/CartesianList.java
@Override public int size() { return axes.size(); } @Override public E get(int axis) { checkElementIndex(axis, size()); int axisIndex = getAxisIndexForProductIndex(index, axis); return axes.get(axis).get(axisIndex); } @Override boolean isPartialView() { return true; }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Thu Nov 30 21:54:06 UTC 2023 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/end2end.mlir
// CHECK-NEXT: %[[CC0:.*]] = "tf.RiscConcat"(%[[ED0]], %[[ED1]]) {axis = 0 : i32} : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> // CHECK-NEXT: %[[ED2:.*]] = "tf.ExpandDims"(%arg2, %[[AXIS]]) : (tensor<2x3xf32>, tensor<i32>) -> tensor<*xf32> // CHECK-NEXT: %[[CC1:.*]] = "tf.RiscConcat"(%[[CC0]], %[[ED2]]) {axis = 0 : i32} : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/optimize_op_order.mlir
%0 = "tfl.dequantize"(%arg0) : (tensor<1000x1000x!quant.uniform<i8:f32, 7.812500e-03>>) -> tensor<1000x1000xf32> %1 = "tfl.gather"(%0, %arg1) {axis = 0 : i32, batch_dims = 0 : i32}: (tensor<1000x1000xf32>, tensor<1x1xi32>) -> tensor<1x1x1000xf32> func.return %1 : tensor<1x1x1000xf32> // CHECK-NEXT: tfl.gather // CHECK-NEXT: tfl.dequantize }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 01 02:06:15 UTC 2022 - 3.6K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape))); xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape))); auto axis = Const(scope_, 0); auto y = Concat(scope_, xs, axis); TensorShape result_shape({9, 2, 5}); RunTest(xs, {shape, shape, shape}, {y}, {result_shape}); } TEST_F(ArrayGradTest, BroadcastToGrad) { TensorShape x_shape({2, 5});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
std::vector<Output>* grad_outputs) { int N; TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "N", &N)); int axis; TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); grad_outputs->reserve(N); auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis)); for (const Output& o : grad_op.output) { grad_outputs->emplace_back(o); } return scope.status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0)