Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 213 for se_shape (0.12 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions_xla_selective_quantization.mlir

          transpose_a = false, transpose_b = false
        } : (tensor<1x10xf32>, tensor<10x10xf32>) -> tensor<1x10xf32>  loc(fused["MatMul:", "test_opt_out"])
        %2 = "tf.Reshape"(%1, %cst_0) : (tensor<1x10xf32>, tensor<2xi32>) -> tensor<?x10xf32> loc(fused["Reshape:", "model/reshape"])
        %3 = "tf.AddV2"(%2, %cst) : (tensor<?x10xf32>, tensor<10xf32>) -> tensor<?x10xf32> loc(fused["AddV2:", "model/add"])
        func.return %3 : tensor<?x10xf32>
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        // Reshapes the R1 length to a scalar.
        auto reshape = builder.create<TF::ReshapeOp>(
            length.getLoc(),
            ArrayRef<Type>{RankedTensorType::get(
                {}, getElementTypeOrSelf(current_size.getType()))},
            ArrayRef<Value>{current_size,
                            cutil::GetR1Const({}, builder, length.getLoc())});
        length.getLength().replaceAllUsesWith(reshape);
      }
      length.erase();
      return success();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.cc

        }
      }
    
      auto ranked_type = mlir::dyn_cast<RankedTensorType>(result_type);
      if (!ranked_type) return UnrankedTensorType::get(builder->getI1Type());
    
      return RankedTensorType::get(ranked_type.getShape(), builder->getI1Type());
    }
    
    Type InferReductionOpType(Value input, Value reduction_indices,
                              BoolAttr keep_dims) {
      Type input_ty = input.getType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_without_identity.pbtxt

    # MLIR:         %[[reshape:.*]] = "tfl.reshape"(%[[conv]], %[[shape]]) : (tensor<1x1x1x186x!quant.uniform<i8:f32, 0.09363494573854933:22>>, tensor<3xi32>)
    # MLIR:         return %[[reshape]] : tensor<1x6x31x!quant.uniform<i8:f32, 0.09363494573854933:22>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_without_identity_4bit.pbtxt

    # MLIR:         %[[reshape:.*]] = "tfl.reshape"(%[[conv]], %[[shape]]) : (tensor<1x1x1x186x!quant.uniform<i8:f32, 0.09363494573854933:22>>, tensor<3xi32>)
    # MLIR:         return %[[reshape]] : tensor<1x6x31x!quant.uniform<i8:f32, 0.09363494573854933:22>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_per_channel.pbtxt

    # MLIR:         %[[reshape:.*]] = "tfl.reshape"(%[[conv]], %[[shape]]) : (tensor<1x1x1x186x!quant.uniform<i8:f32, 0.09363494573854933:22>>, tensor<3xi32>)
    # MLIR:         return %[[reshape]] : tensor<1x6x31x!quant.uniform<i8:f32, 0.09363494573854933:22>>
    # MLIR:       }
    
    # CHECK-LABEL: {
    # CHECK:   version: 3,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 18.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tests/prepare-tf-fake-quant.mlir

      %1 = "tf.Reshape"(%0, %cst_0) : (tensor<1x2xf32>, tensor<2xi64>) -> tensor<2x1xf32>
      %2 = "tf.Reshape"(%0, %cst_0) : (tensor<1x2xf32>, tensor<2xi64>) -> tensor<2x1xf32>
      func.return %1, %2 : tensor<2x1xf32>, tensor<2x1xf32>
    
    // CHECK:  %cst = arith.constant
    // CHECK:  %[[FQ:.*]] = "tf.FakeQuantWithMinMaxVars"(%arg0, %arg1, %arg2)
    // CHECK:  %[[R1:.*]] = "tf.Reshape"(%[[FQ]], %cst)
    // CHECK-SAME: tensor<2x1xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 20.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h

        return operand_ty == result_ty && identity_ty.hasStaticShape() &&
               result_ty.hasStaticShape() &&
               OpTrait::util::staticallyKnownBroadcastable(operand_ty.getShape(),
                                                           identity_ty.getShape());
      };
    
      // Check that we have a constant operand on one side (candidate for identity).
      const bool is_commutative =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.td

    def HasEqualShape : Constraint<CPred<
      "$0.getType().cast<ShapedType>().hasRank() && "
      "$1.getType().cast<ShapedType>().hasRank() && "
      "$0.getType().cast<ShapedType>().getShape() == $1.getType().cast<ShapedType>().getShape()">,
      "Checks if the shapes of tensors are same.">;
    
    // Make the 1D value $0 broadcastable with the shape of $1.
    def MakeOneDimValueBroadcastable : NativeCodeCall<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 03:24:59 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/aot/test_graph_tfadd.pbtxt

        }
      }
      attr {
        key: "dtype"
        value {
          type: DT_INT32
        }
      }
    }
    node {
      name  : "y_reshape"
      op    : "Reshape"
      input : "y_const"
      input : "y_shape"
      attr { key: "T" value { type: DT_INT32 } }
      # Attribute TShape not specified; needs to be set to its default
      # by tfcompile.
    }
    node {
      name  : "x_y_sum"
      op    : "Add"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jul 07 01:22:33 UTC 2017
    - 894 bytes
    - Viewed (0)
Back to top