Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 28 of 28 for in_shape (0.34 sec)

  1. src/cmd/compile/internal/typecheck/subr.go

    			return ir.OCONVNOP, ""
    		}
    		if src.IsShape() || dst.IsShape() {
    			// Conversion between a shape type and one of the types
    			// it represents also needs no conversion.
    			return ir.OCONVNOP, ""
    		}
    	}
    
    	// 3. dst is an interface type and src implements dst.
    	if dst.IsInterface() && src.Kind() != types.TNIL {
    		if src.IsShape() {
    			// Shape types implement things they have already
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Oct 05 19:45:58 UTC 2023
    - 20.2K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/array_grad.cc

      if (op.num_inputs() != 2) {
        return errors::InvalidArgument("BroadcastTo requires 2 inputs");
      }
    
      auto x_shape = Shape(scope, op.input(0));
      auto args = internal::BroadcastGradientArgs(scope, x_shape, op.input(1));
      auto sum_gx = Sum(scope, grad_inputs[0], args.r0);
      grad_outputs->push_back(Reshape(scope, sum_gx, x_shape));
      grad_outputs->push_back(NoGradient());
      return scope.status();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir

      %tl = "tf.TensorListFromTensor"(%arg0, %elem_shape) : (tensor<10x8x9xf32>, tensor<2xi32>) -> tensor<!tf_type.variant<tensor<8x9xf32>>>
      // CHECK: %[[IND_SHAPE:.*]] = "tf.Const"() <{value = dense<[5, 1]> : tensor<2xi32>}> : () -> tensor<2xi32>
      // CHECK: %[[IND_RESHPE:.*]] = "tf.Reshape"(%[[ARG1]], %[[IND_SHAPE]]) : (tensor<5xi32>, tensor<2xi32>) -> tensor<5x1xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 38.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        Type output_type = op.getResult().getType();
        const int64_t rank = operand_type.getRank();
    
        ArrayRef<int64_t> idx_shape(rank);
        TensorType idx_type =
            operand_type.cloneWith(idx_shape, rewriter.getI32Type());
    
        ArrayRef<int64_t> start_idx_i64 = op.getStartIndices();
        ArrayRef<int64_t> limit_idx_i64 = op.getLimitIndices();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/schema/schema_generated.h

      typedef VariantSubTypeBuilder Builder;
      enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
        VT_SHAPE = 4,
        VT_TYPE = 6,
        VT_HAS_RANK = 8
      };
      const ::flatbuffers::Vector<int32_t> *shape() const {
        return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_SHAPE);
      }
      tflite::TensorType type() const {
        return static_cast<tflite::TensorType>(GetField<int8_t>(VT_TYPE, 0));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
  6. src/cmd/compile/internal/types/type.go

    func (t *Type) Noalg() bool      { return t.flags&typeNoalg != 0 }
    func (t *Type) Deferwidth() bool { return t.flags&typeDeferwidth != 0 }
    func (t *Type) Recur() bool      { return t.flags&typeRecur != 0 }
    func (t *Type) IsShape() bool    { return t.flags&typeIsShape != 0 }
    func (t *Type) HasShape() bool   { return t.flags&typeHasShape != 0 }
    
    func (t *Type) SetNotInHeap(b bool)  { t.flags.set(typeNotInHeap, b) }
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Apr 04 14:29:45 UTC 2024
    - 49.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-DAG: [[RNG_SHAPE:%.*]] = mhlo.constant dense<4> : tensor<1xi64>
      // CHECK-DAG: [[RNG_LOWER:%.*]] = mhlo.constant dense<0> : tensor<i32>
      // CHECK-DAG: [[RNG_UPPER:%.*]] = mhlo.constant dense<4> : tensor<i32>
      // CHECK: [[SWAPS:%.*]] = "mhlo.rng"([[RNG_LOWER]], [[RNG_UPPER]], [[RNG_SHAPE]]) <{rng_distribution = #mhlo.rng_distribution<UNIFORM>}>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        Arg<TF_Int64Tensor, [{2-D.  The `indices` of the second `SparseTensor`, size `[nnz, ndims]` Matrix.}]>:$b_indices,
        Arg<TF_NumberTensor, [{1-D.  The `values` of the second `SparseTensor`, size `[nnz]` Vector.}]>:$b_values,
        Arg<TF_Int64Tensor, [{1-D.  The `shape` of the second `SparseTensor`, size `[ndims]` Vector.}]>:$b_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top