- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 204 for dims2 (0.05 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// A sharding with OTHER type may be REPLICATED if: // 'replicate_on_last_tile_dim' is true OR // 'last_tile_dims' is not empty // AND // other than replicated last tile dims, all other dims are not sharded. bool IsOtherReplicatedSharding(const xla::OpSharding& xla_sharding); // Returns whether the sharding is split sharding. i.e. A sharding with OTHER // type but not replicated.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
TF_CAPI_EXPORT extern TF_DataType TF_CheckpointReaderGetVariableDataType( TF_CheckpointReader* reader, const char* name); // Read the shape of a variable and write to `dims` TF_CAPI_EXPORT extern void TF_CheckpointReaderGetVariableShape( TF_CheckpointReader* reader, const char* name, int64_t* dims, int num_dims, TF_Status* status); // Get the number of dimension of a variable TF_CAPI_EXPORT extern int TF_CheckpointReaderGetVariableNumDims(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir
%2 = stablehlo.reshape %1 : (tensor<i32>) -> tensor<1xi32> %3 = stablehlo.concatenate %2, %cst_0, dim = 0 : (tensor<1xi32>, tensor<1xi32>) -> tensor<2xi32> %4 = stablehlo.dynamic_broadcast_in_dim %arg2, %3, dims = [1] : (tensor<3xf32>, tensor<2xi32>) -> tensor<?x3xf32> %5 = stablehlo.add %0, %4 : tensor<?x3xf32> return %5 : tensor<?x3xf32> } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 91.6K bytes - Viewed (0) -
platforms/software/platform-base/src/test/groovy/org/gradle/platform/base/internal/DefaultBinaryNamingSchemeTest.groovy
original.withBinaryType("other").getOutputDirectory(new File(".")) == new File(".", "parent/other/dim1") original.withVariantDimension("dim2").getOutputDirectory(new File(".")) == new File(".", "parent/type/dim1/dim2") original.withRole("role", false).getOutputDirectory(new File(".")) == new File(".", "parent/role/dim1")
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Wed Oct 11 12:16:09 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
EXPECT_TRUE(status.ok()); } absl::StatusOr<std::unique_ptr<Graph>> BuildOpGraphWithOutputShapes() { DataType data_type = DT_INT32; std::initializer_list<int64_t> dims = {2, 3, 4, 5}; Tensor tensor(data_type, TensorShape(dims)); for (int i = 0; i < 2 * 3 * 4 * 5; ++i) { tensor.flat<int32>()(i) = i; } NodeDef node; auto builder = NodeDefBuilder("some_node", "Const")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
// could process the following equation by setting the attributes properly: // abc,cd->abd. // 4. The output should be in the form: [batch dims][lhs dims][rhs dims] bool IsEinsumSupportedByXlaDotV2(StringAttr equation_attr) { StringRef equation = equation_attr.getValue(); if (!absl::StrContains(equation, "->") || !absl::StrContains(equation, ",") ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
android/guava-tests/test/com/google/common/primitives/LongsTest.java
int dim1 = 1 << 16; int dim2 = 1 << 15; assertThat(dim1 * dim2).isLessThan(0); testConcat_overflow(dim1, dim2); } @GwtIncompatible // different overflow behavior; could probably be made to work by using ~~ public void testConcat_overflow_nonNegative() { int dim1 = 1 << 16; int dim2 = 1 << 16; assertThat(dim1 * dim2).isAtLeast(0); testConcat_overflow(dim1, dim2); }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue Feb 20 17:00:05 UTC 2024 - 30K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
FindOperationOfType<DotGeneralOp>(main_fn)->getResult(0); EXPECT_TRUE(HasStaticShape(dot_general_result)); EXPECT_TRUE(HasStaticShapeAtDims(dot_general_result, /*dims=*/{0})); EXPECT_TRUE(HasStaticShapeAtDims(dot_general_result, /*dims=*/{1})); } TEST_F(AttrsAndConstraintsTest, HasStaticShapeFailsWithDynamicShapes) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleDynamic); ASSERT_TRUE(module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
guava-tests/test/com/google/common/primitives/LongsTest.java
int dim1 = 1 << 16; int dim2 = 1 << 15; assertThat(dim1 * dim2).isLessThan(0); testConcat_overflow(dim1, dim2); } @GwtIncompatible // different overflow behavior; could probably be made to work by using ~~ public void testConcat_overflow_nonNegative() { int dim1 = 1 << 16; int dim2 = 1 << 16; assertThat(dim1 * dim2).isAtLeast(0); testConcat_overflow(dim1, dim2); }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue Feb 20 17:00:05 UTC 2024 - 30K bytes - Viewed (0)