- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 205 for dims1 (0.07 sec)
-
tensorflow/c/experimental/ops/array_ops.cc
// shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5] // shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1] // ``` // // This operation requires that: // // `-1-input.dims() <= dim <= input.dims()` // // This operation is related to `squeeze()`, which removes dimensions of // size 1. Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource-device-inference.mlir
%cst = arith.constant dense<3.0> : tensor<32xf32> %cmp = "tf.Less"(%read, %cst) : (tensor<32xf32>, tensor<32xf32>) -> tensor<32xi1> %dims = arith.constant dense<0> : tensor<1xi32> %reduce = "tf.All"(%cmp, %dims) {keep_dims = false} : (tensor<32xi1>, tensor<1xi32>) -> tensor<i1> "tf.Yield"(%reduce) : (tensor<i1>) -> () }, {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 17 16:01:45 UTC 2022 - 18.2K bytes - Viewed (0) -
tensorflow/c/c_api.h
// Returns the shape of the Tensor referenced by `output` in `graph` // into `dims`. `dims` must be an array large enough to hold `num_dims` // entries (e.g., the return value of TF_GraphGetTensorNumDims). // // If the number of dimensions in the shape is unknown or the shape is // a scalar, `dims` will remain untouched. Otherwise, each element of // `dims` will be set corresponding to the size of the dimension. An
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
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) -
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);
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)