- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 137 for cc_ops (0.12 sec)
-
tensorflow/cc/BUILD
hdrs = ["gradients/grad_helper.h"], visibility = ["//tensorflow:internal"], deps = [ ":cc_ops", "//tensorflow/cc:ops", "//tensorflow/cc:scope", ], ) cc_library( name = "linalg_grad", srcs = ["gradients/linalg_grad.cc"], deps = [ ":cc_ops", ":cc_ops_internal", ":grad_helper", ":grad_op_registry", ":gradients",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include <algorithm> #include <cstdint> #include <functional> #include <limits> #include <numeric> #include <string> #include <tuple> #include <type_traits> #include <utility>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
llvm::SmallVector<TF::IfRegionOp, 8> sub_merged_groups; auto begin_if_op_iter = if_ops.begin(); while (begin_if_op_iter != if_ops.end()) { auto current_last_if_op_iter = FindLastIfInSegment(begin_if_op_iter, if_ops, side_effect_analysis); assert(current_last_if_op_iter != if_ops.end()); llvm::SmallVector<TF::IfRegionOp, 8> sub_if_group;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
// it's a constant. Basically, `paddings` tensor in `SpaceToBatch` and `crops` // tensor in `BatchToSpace` must satisfy the following: // paddings[i, 0] = base_paddings[i, 0]. // 0 <= paddings[i, 1] - base_paddings[i, 1] < block_shape[i] // (input_shape[i] + paddings[i, 0] + paddings[i, 1]) % block_shape[i] == 0. // crops[i, 0] = 0. // crops[i, 1] = paddings[i, 1] - base_paddings[i, 1].
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
return true; } // Check if the preceding op is a quantized same-scale op. if (llvm::isa<quantfork::StorageCastOp>(preceding_op)) { auto sc_op = llvm::cast<quantfork::StorageCastOp>(preceding_op); auto sc_arg_type = sc_op.getArg().getType().dyn_cast<TensorType>(); if (sc_arg_type.getElementType().isInteger(8)) { return true; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
TFL::FullyConnectedOp fc_op, PatternRewriter& rewriter) const { // We have to know the shape of the input. auto input = fc_op.getInput(); auto input_type = mlir::dyn_cast_or_null<RankedTensorType>(input.getType()); if (!input_type || !input_type.hasStaticShape()) return failure(); // We have to know the shape of the weight. auto weight = fc_op.getFilter();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize.td
"ArrayRef<APInt>{$0.getValue(), $0.getValue()})">; def BatchToSpaceToBatchToSpaceND : Pat< (TF_BatchToSpaceOp:$src $input, $crops, $block_size), (TF_BatchToSpaceNDOp:$dest $input, (TF_ConstOp (BatchToSpaceBlockSizeToBlockShape $block_size)), $crops), [(IsRank4Tensor $input), (IsRank2Tensor $crops)], [(CopyAttrs $src, $dest)]>; //===----------------------------------------------------------------------===//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:42:28 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
const Tensor& cc_b, Tensor* cc_out, void (*binary_add_func)(TF_OpKernelContext* ctx, TF_Tensor* a, TF_Tensor* b, TF_Tensor* out)) { if (cc_a.dtype() == ::tensorflow::DT_INVALID) { *cc_out = cc_b; return absl::OkStatus(); } if (cc_b.dtype() == ::tensorflow::DT_INVALID) { *cc_out = cc_a; return absl::OkStatus(); } Status status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
// $orig_input_shape CompileTimeConstantOperand<TF::AvgPoolGradOp, 0>, // $block_shape, $crops CompileTimeConstantOperand<TF::BatchToSpaceNDOp, 1, 2>, CompileTimeConstantOperand<TF::BatchToSpaceOp, 1>, // $crops CompileTimeConstantOperand<TF::BincountOp, 1>, // $size CompileTimeConstantOperand<TF::BroadcastArgsOp, 0, 1>, // $s0, $s1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/BUILD
) exports_files([ "ir/tf_generated_ops.td", "ir/tf_op_base.td", "ir/tf_op_interfaces.td", "ir/tf_ops.td", ]) td_library( name = "tensorflow_ops_td_files", srcs = [ "ir/tf_generated_ops.td", "ir/tf_op_base.td", "ir/tf_op_interfaces.td", "ir/tf_ops.td", ], compatible_with = get_compatible_with_portable(), deps = [
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 47.7K bytes - Viewed (0)