- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 75 for shape_n (0.17 sec)
-
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/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/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/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) -
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) -
tensorflow/c/experimental/saved_model/core/tensor_spec.cc
TensorSpec::TensorSpec() : shape_(std::initializer_list<int64_t>()), dtype_(DT_FLOAT) {} TensorSpec::TensorSpec(PartialTensorShape shape, DataType dtype) : shape_(std::move(shape)), dtype_(dtype) {} TensorSpec::TensorSpec(const TensorSpecProto& proto) : shape_(proto.shape()), dtype_(proto.dtype()) {} const PartialTensorShape& TensorSpec::shape() const { return shape_; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 1.3K bytes - Viewed (0) -
test/typeparam/shape1.out
Keith Randall <******@****.***> 1627678827 -0700
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Sat Jul 31 17:03:07 UTC 2021 - 10 bytes - Viewed (0)