- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 35 for output_dims (0.65 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
const ArrayRef<int64_t> output_dims) { if (other_tensor_zp == 0) { return CreateScalarConstValue<int32_t>(builder, loc, 0); } auto shape = mlir::cast<ShapedType>(tensor.getType()); SmallVector<int64_t> non_output_indices; for (int64_t i : llvm::seq<int64_t>(0, shape.getRank())) { if (absl::c_count(output_dims, i) == 0) { non_output_indices.push_back(i); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/c/kernels.cc
return tf_tensor; } TF_Tensor* TF_ForwardInputOrAllocateOutput( TF_OpKernelContext* context, const int* candidate_input_indices, int num_candidate_input_indices, int output_index, const int64_t* output_dims, int output_num_dims, int* forwarded_input, TF_Status* status) { TF_SetStatus(status, TF_OK, ""); auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
<< output_dims[dnums.output_batch_dimension()]); output_dims[dnums.output_feature_dimension()] = kernel_output_feature; DCOMMENT("inferrd output output_feature_dimension is " << output_dims[dnums.output_feature_dimension()]); std::vector<int64_t> input_spatial_dims; llvm::SmallVector<int64_t> window_spatial_dims; for (auto i = 0; i < num_spatial_dims; ++i) { input_spatial_dims.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
TF_Status* s = TF_NewStatus(); int candidate_input_indices[1] = {0}; int forwarded_input; int64_t output_dims[1] = {}; TF_Tensor* output = TF_ForwardInputOrAllocateOutput( /*context=*/ctx, candidate_input_indices, /*num_candidate_input_indices=*/1, /*output_index=*/0, output_dims, /*output_num_dims=*/0, &forwarded_input, /*status=*/s); EXPECT_EQ(TF_OK, TF_GetCode(s));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/c/kernels.h
// -1. TF_CAPI_EXPORT TF_Tensor* TF_ForwardInputOrAllocateOutput( TF_OpKernelContext* context, const int* candidate_input_indices, int num_candidate_input_indices, int output_index, const int64_t* output_dims, int output_num_dims, int* forwarded_input, TF_Status* status); // Allocates a temporary Tensor of the specified type and shape. The // Tensor must not be used after kernel construction is // complete. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1); llvm::SmallVector<int64_t, 4> output_dims = llvm::to_vector<4>(indices_shape); output_dims.insert(output_dims.begin() + axis, depth); Location loc = op.getLoc(); // The iota result is the effective output shape of the computation,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/generate_cpp_main.cc
tensorflow::Flag( "namespace", &name_space_, "Compact C++ namespace, default is 'tensorflow::ops'."), tensorflow::Flag( "output_dir", &output_dir_, "Directory into which output files will be generated."), tensorflow::Flag( "source_dir", &source_dir_, "The tensorflow root directory, e.g. 'tensorflow/' for "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 21 18:51:25 UTC 2021 - 3.4K bytes - Viewed (0) -
platforms/jvm/language-java/src/main/java/org/gradle/api/internal/tasks/compile/CleaningJavaCompiler.java
ImmutableSet.Builder<File> outputDirs = ImmutableSet.builderWithExpectedSize(3); MinimalJavaCompileOptions compileOptions = spec.getCompileOptions(); addDirectoryIfNotNull(outputDirs, spec.getDestinationDir()); addDirectoryIfNotNull(outputDirs, compileOptions.getAnnotationProcessorGeneratedSourcesDirectory()); addDirectoryIfNotNull(outputDirs, compileOptions.getHeaderOutputDirectory());
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Aug 28 11:40:18 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/common/path_config.cc
#include "tensorflow/core/platform/types.h" namespace tensorflow { namespace generator { PathConfig::PathConfig(const string& output_dir, const string& source_dir, const string& api_dir_list, const std::vector<string> op_names) : output_path(output_dir), op_names(op_names) { api_dirs = str_util::Split(api_dir_list, ",", str_util::SkipEmpty());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 09:51:28 UTC 2024 - 2.6K bytes - Viewed (0) -
platforms/core-execution/execution/src/integTest/groovy/org/gradle/internal/execution/MutableUnitOfWorkBuilder.groovy
Map<String, ? extends File> outputDirs, Collection<? extends TestFile> create, InputFingerprinter inputFingerprinter, ExecutionHistoryStore executionHistoryStore ) { this.inputProperties = inputProperties this.inputFiles = inputFiles this.outputFiles = outputFiles this.outputDirs = outputDirs this.create = create
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Wed Nov 22 09:41:28 UTC 2023 - 9.2K bytes - Viewed (0)