- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 24 for add_shape (0.31 sec)
-
tensorflow/compiler/jit/shape_inference.cc
const InferredShape& arg_shape = it->second; shape_inference::InferenceContext* context = shape_refiner->GetContext(n); if (arg_shape.handle_type != DT_INVALID) { shape_inference::ShapeHandle handle; TF_RETURN_IF_ERROR(context->MakeShapeFromPartialTensorShape( arg_shape.handle_shape, &handle));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
guava/src/com/google/common/escape/CharEscaperBuilder.java
@CanIgnoreReturnValue public CharEscaperBuilder addEscape(char c, String r) { map.put(c, checkNotNull(r)); if (c > max) { max = c; } return this; } /** Add multiple mappings at once for a particular index. */ @CanIgnoreReturnValue public CharEscaperBuilder addEscapes(char[] cs, String r) { checkNotNull(r); for (char c : cs) { addEscape(c, r); } return this; }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue Jan 18 20:55:09 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
}; struct VariantSubTypeBuilder { typedef VariantSubType Table; ::flatbuffers::FlatBufferBuilder &fbb_; ::flatbuffers::uoffset_t start_; void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) { fbb_.AddOffset(VariantSubType::VT_SHAPE, shape); } void add_type(tflite::TensorType type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
android/guava/src/com/google/common/escape/CharEscaperBuilder.java
@CanIgnoreReturnValue public CharEscaperBuilder addEscape(char c, String r) { map.put(c, checkNotNull(r)); if (c > max) { max = c; } return this; } /** Add multiple mappings at once for a particular index. */ @CanIgnoreReturnValue public CharEscaperBuilder addEscapes(char[] cs, String r) { checkNotNull(r); for (char c : cs) { addEscape(c, r); } return this; }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue Jan 18 20:55:09 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
absl::StatusOr<TensorShape> GetTensorShapeFromXlaArgument( const XlaArgument& arg) { if (absl::holds_alternative<xla::Shape>(arg.shape)) { TensorShape arg_shape; TF_RETURN_IF_ERROR( XLAShapeToTensorShape(std::get<xla::Shape>(arg.shape), &arg_shape)); return arg_shape; } else { return std::get<TensorShape>(arg.shape); } } Status MaybeRewriteLayoutWithShardedShape( mlir::StringAttr sharding,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
// TODO(kramm): Move this into a separate pass. See b/184944903 xla::Shape old_shape = xla::TypeToShape(t); XLA_Shape old_shape_c = {}; XLA_Shape new_shape_c = {}; TfTpu_ExecutorApiFn *executor = stream_executor::tpu::ExecutorApiFn(); if (!stream_executor::tpu::IsInitialized(executor)) { return failure(); } ApiConverter::ToC(old_shape, &old_shape_c);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
auto client = xla::ClientLibrary::GetOrCreateCompileOnlyClient(platform).value(); tensorflow::tpu::TPUCompileMetadataProto metadata_proto; // Set up an arg per arg_shape with the specified type. for (int i = 0; i < arg_shapes.size(); ++i) { auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind( tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto paddings_value = builder->create<TF::ConstOp>(filter.getLoc(), paddings); std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, filter_shape[1] + pad_w, filter_shape[2], filter_shape[3]}; SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); auto expand_result_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
android/guava/src/com/google/common/escape/Escapers.java
* @return the builder instance * @throws NullPointerException if {@code replacement} is null */ @CanIgnoreReturnValue public Builder addEscape(char c, String replacement) { checkNotNull(replacement); // This can replace an existing character (the builder is re-usable). replacementMap.put(c, replacement); return this; }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Wed Apr 26 20:07:17 UTC 2023 - 10.5K bytes - Viewed (0) -
guava/src/com/google/common/escape/Escapers.java
* @return the builder instance * @throws NullPointerException if {@code replacement} is null */ @CanIgnoreReturnValue public Builder addEscape(char c, String replacement) { checkNotNull(replacement); // This can replace an existing character (the builder is re-usable). replacementMap.put(c, replacement); return this; }
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Wed Apr 26 20:07:17 UTC 2023 - 10.5K bytes - Viewed (0)