Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 35 of 35 for output_shapes (0.26 sec)

  1. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

      Value none = rewriter.create<TFL::NoValueOp>(
          op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr());
    
      Value output_shape =
          CreateCastToInt32(tf_op.getInputSizes(), op->getLoc(), rewriter);
    
      rewriter.replaceOpWithNewOp<TFL::Conv3DTransposeOp>(
          op, tf_op.getType(), output_shape, tf_op.getFilter(),
          tf_op.getOutBackprop(),
          /*bias=*/none, dilation_depth_factor, dilation_height_factor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

      // the shape inference pass is run early in the pass pipeline, shape inference
      // during import is not necessary.
      config.enable_shape_inference = false;
      // Some graphs may require _output_shapes (an unregistered attribute)
      // to override shapes. It is unfortunately not always set correctly so only
      // do it optionally.
      config.unconditionally_use_set_output_shapes =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir

    func.func @testUnimplementedOp() -> (tensor<i32>, tensor<i32>) {
      %0 = arith.constant dense<1> : tensor<i32>
      %1 = arith.constant dense<2> : tensor<i32>
      %2 = "tf.Maximum"(%0, %1) {_output_shapes = ["tfshape$"]} : (tensor<i32>, tensor<i32>) -> tensor<i32>
      %3 = "tf.Minimum"(%0, %1) {random_attr = "hello"} : (tensor<i32>, tensor<i32>) -> tensor<i32>
      func.return %2, %3: tensor<i32>, tensor<i32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 31 23:22:24 UTC 2024
    - 36.7K bytes
    - Viewed (0)
Back to top