- Sort Score
- Result 10 results
- Languages All
Results 41 - 43 of 43 for input_type (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
OpBuilder builder(module); std::vector<Type> input_types = GetValueTypes(inputs); std::vector<Type> output_types = GetValueTypes(outputs); builder.setInsertionPointToEnd(&module.getBodyRegion().back()); func::FuncOp func_op = builder.create<func::FuncOp>( module.getLoc(), name, builder.getFunctionType(input_types, output_types)); func_op.setPrivate(); return func_op; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
} else { // Graphdef import path. module = tensorflow::LoadFromGraphdefOrMlirSource( input_file_name, input_mlir, use_splatted_constant, custom_opdefs, specs, debug_info_file, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, &source_mgr, &context); } // If errors occur, the library call in the above already logged the error // message. So we can just return here.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/c/kernels.cc
auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); CHECK_GE(index, 0); // Crash OK CHECK_LT(index, cc_ctx->num_inputs()); // Crash OK return static_cast<TF_DataType>(cc_ctx->input_dtype(index)); } void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)