- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 45 for NCHW (0.07 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
// LHS transpose permutation must be a NCHW->NHWC permutation. template <typename OpT> void DeferRhsTransposeForBinaryOp(OpT op, PatternRewriter& rewriter) { auto transpose_op = cast<TransposeOp>(op.getOperand(0).getDefiningOp()); Value lhs_pre_transpose = transpose_op.getOperand(); // NCHW -> NHWC for the right-hand side, to match the operand's shape. Value rhs = op.getOperand(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/layout_optimization_layout_assignment_to_nhwc.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
reverse[permutation[i]] = i; } return reverse; } SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to) { if (from == "NHWC" && to == "NCHW") { return {0, 3, 1, 2}; } else if (from == "NCHW" && to == "NHWC") { return {0, 2, 3, 1}; } else { return {}; } } // Shuffle elements in the `attr` according to the permutation. Optional
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/keras_imagenet_main.pbtxt
device: "/job:localhost/replica:0/task:0/device:GPU:0" attr { key: "T" value { type: DT_HALF } } attr { key: "data_format" value { s: "NCHW" } } attr { key: "dilations" value { list { i: 1 i: 1 i: 1 i: 1 } } } attr {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 02:52:54 UTC 2019 - 1.3M bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h
const DenseIntElementsAttr& dense_attr, PatternRewriter& builder); // Returns a NHWC shaped type from an NCHW shaped type op. // For example- Given a Composite op that wraps a core.aten.avg_pool2d, this // returns the return type of the tfl.average_pool_2d emitted. Note that the // aten.avg_pool2d works with the NCHW layout while tfl.average_pool_2d assumes // NHWC. ShapedType GetNhwcReturnTypeFromNchw(Operation* old_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/keras_imagenet_main_graph_mode.pbtxt
device: "/job:localhost/replica:0/task:0/device:GPU:0" attr { key: "T" value { type: DT_FLOAT } } attr { key: "data_format" value { s: "NCHW" } } attr { key: "dilations" value { list { i: 1 i: 1 i: 1 i: 1 } } } attr {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 02:52:54 UTC 2019 - 1.1M bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/conv_2d_nchw.pbtxt
op: "Conv2D" input: "input" input: "conv_net_2d/conv_2d_0/w/read" attr { key: "T" value { type: DT_FLOAT } } attr { key: "data_format" value { s: "NCHW" } } attr { key: "dilations" value { list { i: 1 i: 1 i: 1 i: 1 } } } attr {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 03 03:26:13 UTC 2021 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
inline constexpr StringRef kQuantizationMethodAttr = "_quantization_method"; // Permutation from the NHWC tensor format to NCHW. This is an inverse // permutation of `kNchwToNhwcPermutation`. inline constexpr std::array<int64_t, 4> kNhwcToNchwPermutation = {0, 3, 1, 2}; // Permutation from the NCHW tensor format to NHWC. This is an inverse // permutation of `kNchwToNhwcPermutation`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// Also FusedBatchNorm in training mode prefers NCHW data format. Check if all // users can efficiently use NHWC data format? if (one_by_one && trivial_strides && trivial_dilations) { return "NHWC"; } // If filter spatial dimensions are unknown or not 1x1 we prefer NCHW, because // it's the fastest option on NVIDIA GPUs with cuDNN library support. return "NCHW"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/nn_ops.cc
// // Description: // It accumulates all the values from out_backprop into the feature dimension. // For NHWC data format, the feature dimension is the last. For NCHW data // format, the feature dimension is the third-to-last. Status BiasAddGrad(AbstractContext* ctx, AbstractTensorHandle* const out_backprop,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.9K bytes - Viewed (0)