- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 442 for pushBack (0.19 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc
if (inference_type == QUANTIZED_INT8) { all_device_inference_types.push_back({device, QUANTIZED_INT8}); } else if (inference_type == QUANTIZED_UINT8) { all_device_inference_types.push_back({device, QUANTIZED_UINT8}); } // We will alway enable float. all_device_inference_types.push_back({device, FLOAT}); } return all_device_inference_types; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/flatbuffer_compatibility_test.cc
const std::string &contents) { std::vector<const char *> include_directories; auto local_include_directory = flatbuffers::StripFileName(filename); include_directories.push_back(local_include_directory.c_str()); include_directories.push_back(nullptr); if (!parser->Parse(contents.c_str(), include_directories.data(), filename.c_str())) { fprintf(stderr, "Failed to parse flatbuffer schema '%s'\n",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 14:28:27 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
outputs_replaced.push_back(user.getResult()); output_types.push_back(user.getType()); } else if (isa<IntegerType>(result_element_type)) { // If the result is an integer tensor, then it doesn't require the // dequantize op in the pattern. outputs_replaced.push_back(result); output_types.push_back(result.getType()); } else { return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
values_for_results.push_back(output); } } } return values_for_results; } // Compute signature for raised func from arugments and outputs of // Operation partition. llvm::SmallVector<Type> TypesFromValues( const llvm::SmallVector<Value>& values) { llvm::SmallVector<Type> types; for (auto value : values) { types.push_back(value.getType()); } return types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
if (rhs_batch_idx.contains(i)) { if (rhs_only_batch) { rhs_eq.push_back(c); lhs_batch_dims.push_back(c); c++; } else { rhs_eq.push_back(lhs_batch_dims[batch_trace_idx]); batch_trace_idx++; } } else if (rhs_contract_idx.contains(i)) { absl::StrAppend(&rhs_eq, std::string(1, lhs_contract_dims[contract_trace_idx]));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
rmax = std::max(std::abs(rmin), std::abs(rmax)); rmin = -rmax; } TensorRangeSanityCheck(op, rmin, rmax); mins.push_back(rmin); maxs.push_back(rmax); } quant_type = quantfork::fakeQuantAttrsToType( op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range, expressed, is_signed);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
reshaped_shape_ints.push_back(outer_shape_ints[i]); reshaped_shape_ints.push_back(block_shape_ints[i]); } for (int64_t i = 1 + block_rank; i < input_rank; ++i) { reshaped_shape_vals.push_back(padded_shape_splits[i]); reshaped_shape_ints.push_back(padded_shape[i]); } auto reshaped_shape = ValuesToRank1(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
op->getLoc(), unconstrainted_tensor_type, op->getOperand(i)); variadic_operands.push_back(casted); } auto build_list_op = builder.create<BuildListOp>( op->getLoc(), list_type, variadic_operands); new_operands.push_back(build_list_op.getOut()); } else { auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.cc
TF_RET_CHECK(sit != shape_info.end()) << "Missing shape information for node " << node->name(); std::vector<PartialTensorShape> shapes; for (const auto& output : sit->second) shapes.push_back(output.shape); auto it = expected_shapes.find(node->name()); if (it != expected_shapes.end()) { if (!PartialTensorShapeUtils::AreIdentical(shapes, it->second)) { return errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
SmallVector<int64_t, 8> perm0_values; for (const auto &value : perm0.getValues<APInt>()) perm0_values.push_back(value.getSExtValue()); SmallVector<int64_t, 8> perm1_values; for (const auto &value : perm1.getValues<APInt>()) perm1_values.push_back(value.getSExtValue()); for (int i = 0; i < perm0_values.size(); ++i) { if (perm0_values[perm1_values[i]] != i) return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0)