- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 1,825 for Auto (0.04 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
auto x = squared_diff_op.getLhs(); auto y = squared_diff_op.getRhs(); auto x_type = mlir::dyn_cast<RankedTensorType>(x.getType()); auto y_type = mlir::dyn_cast<RankedTensorType>(y.getType()); if (!x_type || !y_type) return failure(); if (x_type.getShape() != y_type.getShape()) return failure(); auto result_type = squared_diff_op.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
Scope root = Scope::NewRootScope().ExitOnError(); auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT, ops::Placeholder::Shape({2, 3})); auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT, ops::Placeholder::Shape({3})); auto c = ops::Placeholder(root.WithOpName("C"), DT_FLOAT); auto d = ops::Add(root.WithOpName("D"), a, b); auto e = ops::Add(root.WithOpName("E"), d, c);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
auto body = while_op.body_function(); auto cond = while_op.cond_function(); auto grads = AccessedGradients({body, cond}, module); auto ta_arg_buffer_type = [&](int64_t index) -> Type { auto it = stats->find(while_op.getOperand(index)); if (it == stats->end()) return nullptr; return it->getFirst().getType(); }; auto ta_accumulate_on_write = [&](int64_t index) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
TensorShape y_shape({2}); auto y = GatherNd(scope_, x, indices); RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, GatherNdGrad_SliceIndexing) { TensorShape shape({2, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); auto indices = Const(scope_, {{1}, {0}}); auto y = GatherNd(scope_, x, indices); RunTest(x, shape, y, shape); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.h
} template <typename OpTy> void PrintExecuteOpFuncAttribute(mlir::OpAsmPrinter &p, OpTy op) { auto op_func_attrs = op.getOpFuncAttrs(); if (!op_func_attrs.empty()) { auto print_key_value = [&](mlir::Attribute attr) { auto key_value = mlir::cast<mlir::ArrayAttr>(attr).getValue(); auto key = key_value[0]; auto value = key_value[1]; p << mlir::cast<mlir::StringAttr>(key).getValue(); p << " = ";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto one_val = DenseElementsAttr::get(type, kOne); auto one_cst = rewriter.create<stablehlo::ConstantOp>(rewriter.getUnknownLoc(), one_val); auto half_val = DenseElementsAttr::get(type, kHalf); auto half_cst = rewriter.create<stablehlo::ConstantOp>(one_cst.getLoc(), half_val); auto one_over_root2_val = DenseElementsAttr::get(type, kOneOverRoot2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
attr = cst.getValue(); } else if (auto cst = dyn_cast<tfl::QConstOp>(inst)) { attr = cst.getValue(); } else if (auto cst = dyn_cast<mlir::stablehlo::ConstantOp>(inst)) { attr = cst.getValue(); } else if (auto cst = dyn_cast<mlir::vhlo::ConstantOpV1>(inst)) { mlir::VhloToStablehloTypeConverter vhlo_type_converter;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
decomposed_partitioned_call_callees) { auto body = while_op.body_function(); llvm::SmallDenseMap<Value, Value> body_map; auto find_arg_stack_type = [&](int64_t index) -> std::optional<Type> { auto it = data_var_to_size_var.find(while_op.getOperand(index)); if (it == data_var_to_size_var.end()) return std::nullopt; return it->getFirst().getType(); };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
auto op = cast<ResizeNearestNeighborOp>(src_op); auto loc = op.getLoc(); auto result_ty = mlir::cast<ShapedType>(op.getType()); auto input = op.getImages(); auto input_ty = mlir::cast<ShapedType>(input.getType()); auto input_element_ty = input_ty.getElementType(); auto out_size = op.getSize(); auto out_size_ty = mlir::cast<ShapedType>(out_size.getType());
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/cc/gradients/math_grad_test.cc
auto x2 = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x2_shape)); auto y = Sub(scope_, x1, x2); RunTest({x1, x2}, {x1_shape, x2_shape}, {y}, {x1_shape}); } TEST_F(NaryGradTest, Mul) { TensorShape x1_shape({3, 2, 5}); TensorShape x2_shape({2, 5}); auto x1 = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x1_shape)); auto x2 = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x2_shape));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 36K bytes - Viewed (0)