- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 113 for const_0d (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
IslandOp GetDummyConstant(OpBuilder builder, ShapedType const_type, Location loc) { DenseIntElementsAttr val = DenseIntElementsAttr::get(const_type, 1); auto const_op = builder.create<TF::ConstOp>(loc, val); auto const_island = CreateIsland(const_op, {}, builder); return const_island; } // Rewrites the while op with extra chaining operands and results. Uses a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
bool isSupportedAxis(mlir::Value value, int64_t rank) const { auto const_op = dyn_cast_or_null<mlir::arith::ConstantOp>(value.getDefiningOp()); if (!const_op) { return false; } auto axes = dyn_cast<DenseIntElementsAttr>(const_op.getValueAttr()); if (!axes || axes.getNumElements() != 1) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
// CHECK-NEXT: %[[CAST_ZERO:.*]] = "tf.Cast"(%[[ZERO_SCALAR]]) : (tensor<i32>) -> tensor<f32> // CHECK-NEXT: %[[CONST10:.*]] = "tf.Const"() <{value = dense<10> : tensor<1xi32>}> : () -> tensor<1xi32> // CHECK-NEXT: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[CAST_ZERO]], %[[CONST10]]) : (tensor<f32>, tensor<1xi32>) -> tensor<10xf32> // CHECK-NEXT: %[[BUFFER:.*]] = "tf.MlirLocalVarOp"() : () -> tensor<!tf_type.resource<tensor<10xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen.cc
*op_header_guard = ToGuard(op_header); const string cc_header = strings::StrCat( R"include(// This file is MACHINE GENERATED! Do not edit. #include "tensorflow/cc/ops/const_op.h" )include", "#include \"", op_header, "\"\n", namespace_begin); const string filename = GetFilename(dot_h_fname); const string doxygen = strings::StrCat("/// @defgroup ", filename, " ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 17:22:47 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass_test.cc
#include "tensorflow/compiler/jit/partially_decluster_pass.h" #include "absl/memory/memory.h" #include "tensorflow/cc/framework/ops.h" #include "tensorflow/cc/ops/array_ops.h" #include "tensorflow/cc/ops/const_op.h" #include "tensorflow/cc/ops/control_flow_ops_internal.h" #include "tensorflow/cc/ops/function_ops.h" #include "tensorflow/cc/ops/sendrecv_ops.h" #include "tensorflow/cc/ops/standard_ops.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
} auto const_op = rewriter.create<TF::ConstOp>(loc, new_type, tensor_proto_attr); // Add scast op to match quantize -> composition pattern. The added scast // is then removed by canonicalization. ([scast - scast] -> []) auto scast_op = rewriter.create<quantfork::StorageCastOp>( loc, tensor_qtype, const_op.getOutput()); q_op->replaceAllUsesWith(scast_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/testdata/outside_compilation.mlir
%0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 19 20:19:45 UTC 2023 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_space_to_depth_pass.mlir
// CHECK: %[[CONST0:.*]] = "tf.Const"() <{value = dense< // CHECK-SAME: [4, 4, 2, 2, 3, 64] // CHECK: %[[RESHAPE0:.*]] = "tf.Reshape"(%[[BACKPROP:.*]], %[[CONST0:.*]]) : (tensor<4x4x12x64xf32>, tensor<6xi64>) -> tensor<4x4x2x2x3x64xf32> // CHECK: %[[CONST1:.*]] = "tf.Const"() <{value = dense<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 37.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
def UndoBroadcastConvBiasAdd : Pat< (TFL_AddOp $lhs, (Arith_ConstantOp:$const_op $bias), TFL_AF_None), (TFL_AddOp $lhs, (Arith_ConstantOp (FlattenTo1D $bias)), TFL_AF_None), [(AnyStaticShapeTensor $lhs), (IsLastDimEqualToNumElements $bias, $bias), (HasOneUse $const_op), (HasRankAtMost<4> $bias), (HasRankAtLeast<2> $bias), (IsDefinedByConv2DOp $lhs)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
{}); } TEST_F(CApiFunctionTest, UsingOneOutputOfSplit) { /* * feed * | * +---------+---+ * | const0 | | * | | | | * | v / | * | split | * | | | | | * | v | v | * | | |
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0)