- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 205 for dims1 (0.05 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
for (int i = 0; i < n; ++i) { result.push_back(i + begin); } return result; }; // We only support the mhlo.dot style input layouts, i.e.: // lhs: [batch, other dims, contract dims] // rhs: [batch, contract dims, other dims] auto dim_nums = dot.getDotDimensionNumbers(); int batch_dims_count = dim_nums.getLhsBatchingDimensions().size(); int contracting_dims_count = dim_nums.getLhsContractingDimensions().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/c/eager/unified_api_test.cc
PartialTensorShape shape; TF_RETURN_IF_ERROR(inputs[0]->Shape(&shape)); if (shape.dims() != 2) { return errors::InvalidArgument( "Tensor expected to have rank 2 found rank: ", shape.dims()); } int64_t dim_sizes[] = {2, 4}; for (int i = 0; i < shape.dims(); i++) { if (shape.dim_size(i) != dim_sizes[i]) { return errors::InvalidArgument("Dim ", i, " expected to be of size ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/tf_tensor.h
// same data buffer. // // This call requires that the `from` tensor and the given type and shape (dims // and num_dims) are "compatible" (i.e. they occupy the same number of bytes). // Specifically, given from_type_size = TF_DataTypeSize(TF_TensorType(from)): // // ShapeElementCount(dims, num_dims) * TF_DataTypeSize(type) // // must equal // // TF_TensorElementCount(from) * from_type_size //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 16:40:30 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/c/eager/tfe_tensor_debug_info_internal.h
#include <vector> #include "tensorflow/core/platform/types.h" struct TFE_TensorDebugInfo { explicit TFE_TensorDebugInfo(const std::vector<int64_t>& dims) : dev_dims(dims) {} // Fully-padded, minor-to-major. std::vector<int64_t> dev_dims; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir
// CHECK: %[[BROADCAST_0:.*]] = stablehlo.broadcast_in_dim %[[CONST_0]], dims = [3] : (tensor<8xf32>) -> tensor<1x1x2x8xf32> // CHECK: %[[MUL:.*]] = stablehlo.multiply %[[ARG]], %[[BROADCAST_0]] : tensor<1x1x2x8xf32> // CHECK: %[[BROADCAST_1:.*]] = stablehlo.broadcast_in_dim %[[CONST_1]], dims = [3] : (tensor<8xf32>) -> tensor<1x1x2x8xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
return absl::OkStatus(); } std::vector<int64_t> dims(num_dims, kUnknownDim); TF_GraphGetTensorShape(graph_, output_, reinterpret_cast<int64_t*>(dims.data()), num_dims, &status); TF_RETURN_IF_ERROR(StatusFromTF_Status(&status)); TF_RETURN_IF_ERROR(tensorflow::TensorShapeUtils::MakeShape(dims, shape)); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
float data[], int64_t dims[], int num_dims); // Get a Matrix TensorHandle with given float values and dimensions TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[], int64_t dims[], int num_dims); // Get a Matrix TensorHandle with given int values and dimensions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K bytes - Viewed (0) -
platforms/jvm/platform-jvm/src/test/groovy/org/gradle/api/internal/tasks/DefaultSourceSetTest.groovy
sourceSet.compiledBy('a') def dirs1 = fileCollectionFactory.configurableFiles() dirs1.builtBy('b') sourceSet.output.dir(dirs1) assertThat(dependencies.getDependencies(null)*.name as Set, equalTo(['a', 'b'] as Set)) dirs1.builtBy('c') assertThat(dependencies.getDependencies(null)*.name as Set, equalTo(['a', 'b', 'c'] as Set))
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Sep 28 15:09:49 UTC 2023 - 10.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
absl::Span<const int64_t> dims, double abs_error) { TF_Tensor* analytical_tensor; auto s = GetValue(t, &analytical_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); int64_t num_elem_analytical = 1; auto num_dims_analytical = TF_NumDims(analytical_tensor); ASSERT_EQ(dims.size(), num_dims_analytical); for (int j = 0; j < num_dims_analytical; j++) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_type_test.cc
namespace tensorflow { namespace { std::string ConvertToMlirString(const std::vector<int64_t>& dims, bool unknown_rank, DataType dtype) { TensorShapeProto shape; shape.set_unknown_rank(unknown_rank); for (int64_t dim : dims) { shape.add_dim()->set_size(dim); } mlir::MLIRContext context; mlir::Builder b(&context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.2K bytes - Viewed (0)