- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 451 for dim2 (0.15 sec)
-
tensorflow/c/experimental/saved_model/core/constant_loading_test.cc
EXPECT_EQ(revived_tensor->Type(), expected.dtype()); EXPECT_EQ(revived_tensor->NumElements(), expected.NumElements()); EXPECT_EQ(revived_tensor->NumDims(), expected.dims()); for (int i = 0; i < expected.dims(); ++i) { EXPECT_EQ(revived_tensor->Dim(i), expected.dim_size(i)); } testing::CheckBufferDataIsEqual(expected.dtype(), expected.NumElements(), revived_tensor->Data(), expected.data());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_type.cc
shape->reserve(input_shape.dims()); for (const auto& d : input_shape) { shape->push_back(d.size == kTFDynamicSize ? ShapedType::kDynamic : d.size); } } Status ConvertToMlirShape(const TensorShapeProto& input_shape, llvm::SmallVectorImpl<int64_t>* shape) { shape->reserve(input_shape.dim_size()); auto& dims = input_shape.dim(); for (auto& d : dims) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
const int64_t rank = type.getRank(); for (const int64_t dim : batch_dimensions) { batch_dimensions_.axes.push_back(dim); batch_dimensions_.sizes.push_back(type.getDimSize(dim)); } for (const int64_t dim : contracting_dimensions) { contracting_dimensions_.axes.push_back(dim); contracting_dimensions_.sizes.push_back(type.getDimSize(dim)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// the same because all strides are 1). low_padding[dim] = conv_grad_spatial_dim.pad_before - orig_padding[dim].first; high_padding[dim] = conv_grad_spatial_dim.pad_after - orig_padding[dim].second; // Update `out_grad_shape` to result shape of following `PadOp`. out_grad_shape[dim] = low_padding[dim] + high_padding[dim] +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (1) -
tensorflow/c/c_api_experimental.cc
std::vector<DimensionHandle> dims; const TF_ShapeAndType& input_shape = input_shapes->items[i]; if (input_shape.num_dims == InferenceContext::kUnknownRank) { c.SetInput(i, c.UnknownShape()); continue; } dims.reserve(input_shape.num_dims); for (int j = 0; j < input_shape.num_dims; ++j) { dims.push_back(c.MakeDim(input_shape.dims[j])); } c.SetInput(i, c.MakeShape(dims));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/add_dump_tensor_op_stablehlo.mlir
%3 = shape.shape_of %2 : tensor<?x2xf32> -> tensor<2xindex> %4 = stablehlo.dynamic_broadcast_in_dim %arg2, %3, dims = [1] : (tensor<2xf32>, tensor<2xindex>) -> tensor<?x2xf32> %5 = stablehlo.add %2, %4 : tensor<?x2xf32> %6 = stablehlo.clamp %0, %5, %1 : (tensor<f32>, tensor<?x2xf32>, tensor<f32>) -> tensor<?x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 18K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
const int num_dims = 2; int64_t* dims = new int64_t[num_dims]; int64_t num_elements = 1; dims[0] = batch_size; dims[1] = 1; for (int64_t i = 0; i < num_dims; ++i) { num_elements *= dims[i]; } TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims, sizeof(TF_TString) * num_elements); delete[] dims;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
// $element_shape, $max_num_elements CompileTimeConstantOperand<TF::EmptyTensorListOp, 0, 1>, CompileTimeConstantOperand<TF::ExpandDimsOp, 1>, // $dim CompileTimeConstantOperand<TF::FillOp, 0>, // $dims CompileTimeConstantOperand<TF::GatherV2Op, 2>, // $axis CompileTimeConstantOperand<TF::IRFFT2DOp, 1>, // $fft_length
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
platforms/core-runtime/files/src/test/groovy/org/gradle/internal/file/FileHierarchySetTest.groovy
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue May 28 13:09:38 UTC 2024 - 12.6K bytes - Viewed (0)