- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 68 for shape_a (0.13 sec)
-
tensorflow/compiler/jit/partially_decluster_pass_test.cc
tensorflow::Scope s = tensorflow::Scope::NewRootScope(); Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, ops::Placeholder::Attrs{}); Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, ops::Placeholder::Attrs{}); Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
const TF_TensorSpec* tensor_spec_a = TF_SignatureDefParamTensorSpec(param_a); const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a); // Input "a" is a scalar, float32 tensor EXPECT_EQ("a", std::string(TF_SignatureDefParamName(param_a))); EXPECT_EQ(TF_FLOAT, TF_TensorSpecDataType(tensor_spec_a)); EXPECT_EQ(0, TF_ShapeDims(shape_a)); const TF_SignatureDefParam* param_b = TF_SignatureDefParamListGet(args, 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
// Gets the broadcasted output shape for tf.BatchMatMulV2Op. `shape_x` is the // shape of op's first/left-hand-side operand and `shape_y` is the shape of // op's second/right-hand-side operand. const auto get_broadcasted_shape = [&](ArrayRef<int64_t> shape_x, ArrayRef<int64_t> shape_y, SmallVectorImpl<int64_t>& result_shape) { if (shape_x.size() < 2 || shape_y.size() < 2) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
shape_before.take_back(shape_c.size()) != shape_c) { return failure(); } ArrayRef<int64_t> shape_y2 = shape_before.drop_front(shape_b.size()).drop_back(shape_c.size()); // No need to check %dot; dot_general verifier ensures correct shapes. // Extract Z from %dot. ArrayRef<int64_t> shape_z = dot.getType().getShape().drop_front(shape_b.size() + shape_y2.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// CHECK-NEXT: %[[SHAPE_0:.*]] = "tf.Const"() // CHECK: {ici_weight_distribution_mlir_bridge_marker = true} // CHECK-NEXT: %[[FULL_0:.*]] = "tf.Fill"(%[[SHAPE_0]], %[[ELEM_0]]) {ici_weight_distribution_mlir_bridge_marker = true} // CHECK: %[[ELEM_1:.*]] = "tf.Const"() // CHECK: {ici_weight_distribution_mlir_bridge_marker = true} // CHECK-NEXT: %[[SHAPE_1:.*]] = "tf.Const"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.td
(HasNoUseOf:$root__3), (HasNoUseOf:$root__4), (HasNoUseOf:$root__5)]>; class HasEqualElementSize<list<int> shape_1, list<int> shape_2> : Constraint< CPred<"quant::HasEqualElementSize($0, $1," "llvm::ArrayRef<int>({" # !interleave(shape_1, ", ") # "})," "llvm::ArrayRef<int>({" # !interleave(shape_2, ", ") # "}))">, "Checks if the given dimensions contain the same number of elements.">; def ReshapableTo1DTensor : Constraint<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:24:59 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
{ GraphDefBuilder shape2(GraphDefBuilder::kFailImmediately); Node* key_constant = KeyPlaceholder("F1", shape2.opts()); Node* recv1 = RecvAtHost( ops::NodeOut(key_constant, 0), "F1", "F1", "O1", {DT_FLOAT, DT_FLOAT}, shape2.opts().WithAttr(kXlaHasHostTransferAttrName, true)); Node* e = Binary(ops::NodeOut(recv1, 0), ops::NodeOut(recv1, 1), shape2.opts()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir
// CHECK-DAG: %[[SHAPE1:.*]] = "tf.Const"() <{value = dense<[1, 32, 32, 16]> : tensor<4xi64>}> // CHECK: %[[SHAPE2:.*]] = "tf.Shape"(%arg2) : (tensor<*xf32>) -> tensor<?xi64> %0:3 = "tf.ShapeN"(%arg0, %arg1, %arg2) : (tensor<f32>, tensor<1x32x32x16xf32>, tensor<*xf32>) -> (tensor<0xi64>, tensor<4xi64>, tensor<?xi64>) // CHECK: return %[[SHAPE0]], %[[SHAPE1]], %[[SHAPE2]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
test/typeparam/shape1.go
Matthew Dempsky <******@****.***> 1646087539 -0800
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Mar 01 19:45:34 UTC 2022 - 855 bytes - Viewed (0)