Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 36 for dim_idx (0.1 sec)

  1. tensorflow/compiler/mlir/lite/utils/utils.h

      SmallVector<int32_t> shape;
      shape.reserve(output_shape.size());
    
      bool needs_truncation = true;
      for (size_t dim_idx = 0; dim_idx < output_shape.size(); ++dim_idx) {
        int64_t dim = output_shape[dim_idx];
        if (truncate && needs_truncation && dim == 1) {
          continue;
        } else if (needs_truncation && dim != 1) {
          needs_truncation = false;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

      }
    
      const int64_t rank = input_type.getRank();
    
      IntegerAttr dim_int = ExtractSingleElementAsInteger(split_dim_attr);
    
      // "axis" operand could be a i64 tensor. Resolve it here.
      IntegerAttr dim_i32;
      if (failed(ConvertToI32Attr(dim_int, &dim_i32))) return failure();
    
      int dim = dim_i32.getInt();
      if (dim < 0) dim += rank;
    
      const SmallVector<Value, 4>& slice_outputs = SliceOutputs(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  3. src/syscall/fs_js.go

    		return 0, err
    	}
    	if f.entries == nil {
    		return 0, EINVAL
    	}
    
    	n := 0
    	for f.dirIdx < len(f.entries) {
    		entry := f.entries[f.dirIdx]
    		l := 2 + len(entry)
    		if l > len(buf) {
    			break
    		}
    		buf[0] = byte(l)
    		buf[1] = byte(l >> 8)
    		copy(buf[2:], entry)
    		buf = buf[l:]
    		n += l
    		f.dirIdx++
    	}
    
    	return n, nil
    }
    
    func setStat(st *Stat_t, jsSt js.Value) {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu May 11 18:19:17 UTC 2023
    - 10.8K bytes
    - Viewed (0)
  4. tensorflow/c/tf_tensor.cc

          dims, num_dims);
    }
    
    int TF_NumDims(const TF_Tensor* t) { return t->tensor->NumDims(); }
    
    int64_t TF_Dim(const TF_Tensor* t, int dim_index) {
      return t->tensor->Dim(dim_index);
    }
    
    size_t TF_TensorByteSize(const TF_Tensor* t) { return t->tensor->ByteSize(); }
    
    void* TF_TensorData(const TF_Tensor* t) { return t->tensor->Data(); }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 21:57:32 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  5. cmd/endpoint.go

    			if err = checkCrossDeviceMounts(eps); err != nil {
    				return nil, setupType, config.ErrInvalidErasureEndpoints(nil).Msg(err.Error())
    			}
    
    			for diskIdx := range eps {
    				eps[diskIdx].SetPoolIndex(poolIdx)
    				eps[diskIdx].SetSetIndex(setIdx)
    				eps[diskIdx].SetDiskIndex(diskIdx)
    			}
    
    			endpoints = append(endpoints, eps...)
    		}
    
    		if len(endpoints) == 0 {
    Registered: Sun Jun 16 00:44:34 UTC 2024
    - Last Modified: Fri May 24 23:05:23 UTC 2024
    - 34.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

        tensor_tile_rank--;
      }
    
      std::map<int, int> dimension_to_splits_map;
      for (int dim_index = 0; dim_index < tensor_tile_rank; ++dim_index) {
        if (sharding.tile_assignment_dimensions(dim_index) > 1) {
          dimension_to_splits_map.emplace(
              dim_index, sharding.tile_assignment_dimensions(dim_index));
        }
      }
    
      if (dimension_to_splits_map.empty()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  7. src/cmd/covdata/dump.go

    	if d.cmd == debugDumpMode {
    		d.cm.ResetModeAndGranularity()
    	}
    }
    
    func (d *dstate) BeginCounterDataFile(cdf string, cdr *decodecounter.CounterDataReader, dirIdx int) {
    	dbgtrace(2, "visit counter data file %s dirIdx %d", cdf, dirIdx)
    	if d.cmd == debugDumpMode {
    		fmt.Printf("data file %s", cdf)
    		if cdr.Goos() != "" {
    			fmt.Printf(" GOOS=%s", cdr.Goos())
    		}
    		if cdr.Goarch() != "" {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue May 23 11:36:57 UTC 2023
    - 11.2K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims(
        TFE_TensorDebugInfo* debug_info);
    
    // Returns the number of elements in dimension `dim_index`.
    // Tensor representation on device can be transposed from its representation
    // on host. The data contained in dimension `dim_index` on device
    // can correspond to the data contained in another dimension in on-host
    // representation. The dimensions are indexed using the standard TensorFlow
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc

          CompileTimeConstantOperand<TF::XlaReduceWindowOp, 2, 3, 4, 5, 6>,
          // $dim_index
          CompileTimeConstantOperand<TF::XlaRemoveDynamicDimensionSizeOp, 1>,
          // $window_dimensions, $window_strides, $padding
          CompileTimeConstantOperand<TF::XlaSelectAndScatterOp, 1, 2, 3>,
          CompileTimeConstantOperand<TF::XlaSetBoundOp, 1>,  // $bound
          // $dim_index
          CompileTimeConstantOperand<TF::XlaSetDynamicDimensionSizeOp, 1>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api.cc

      return num_elements;
    }
    
    int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index,
                                TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return -1;
      }
    
      int64_t dim = -1;
      status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim);
      return dim;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top