- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 788 for shake (0.05 sec)
-
tensorflow/c/c_api_experimental.cc
ShapeHandle shape_handle = c.output(i); TF_ShapeAndType& shape = output_shapes_result->items[i]; shape.num_dims = c.Rank(shape_handle); if (shape.num_dims == InferenceContext::kUnknownRank) { shape.dims = nullptr; continue; } shape.dims = new int64_t[shape.num_dims]; for (size_t j = 0; j < shape.num_dims; ++j) { shape.dims[j] = c.Value(c.Dim(shape_handle, j)); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
// in a `shape` shaped tensor. Assumes row-major order. `indices` and `shape` // should have the same size. // Example: Index (2, 3) of a (4, 5)-shaped tensor has the contiguous offset of // 2 * 5 + 3 = 13. int64_t GetContiguousOffset(const ArrayRef<int64_t> indices, const ArrayRef<int64_t> shape) { int64_t contiguous_offset = 0; int64_t base_offset = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); if (!shape.unknown_rank()) { TF_RETURN_IF_ERROR(operation->SetAttrShape( "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()), shape.dims())); } int num_outputs = 1; std::vector<AbstractTensorHandle*> outputs(num_outputs); TF_RETURN_IF_ERROR(operation->Execute(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
xla_tensor->WaitForDefinitionEventOnStream(device_to_host_stream.get()); // Transfer manager requires the shape of the shaped buffer to be the same as // literal shape except for the layout. Set the literal to use xla_tensor's // shape as it is derived from the cpu_tensor's shape using // shape_representation_fn_. xla::MutableBorrowingLiteral literal; TF_CHECK_OK(HostTensorToMutableBorrowingLiteral(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} // This assumes that the bias is of shape NxCx1x1 and doesn't require transpose // Its corresponding constraint is optimize_patterns.td:IsBiasShape() ElementsAttr ReshapeNCHWBiasToNHWC(Value v, Attribute a) { auto elements = mlir::cast<DenseElementsAttr>(a); auto shape = mlir::cast<ShapedType>(v.getType()).getShape(); if (shape.size() != 4 || shape[2] != 1 || shape[3] != 1) return elements;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/unroll_batch_matmul.pbtxt
node { name: "Placeholder" op: "Placeholder" attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "shape" value { shape { dim { size: 2 } dim { size: 5 } dim { size: 3 } } } } } node {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/optional_input.json
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/cc/framework/gradient_checker.cc
for (int x_idx = 0; x_idx < x_num; x_idx++) { if (x_shapes[x_idx] != dxout[x_idx].shape()) { return errors::Internal("Gradient for input ", x_idx, " expected shape ", x_shapes[x_idx].DebugString(), " but was ", dxout[x_idx].shape().DebugString()); } const int64_t x_size = x_shapes[x_idx].num_elements();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
// Eg 1. An update in canonical form: // * indices shape(A,B,C) // * updates shape(A,B,D,E,F) // Then: // * D,E,F are the update window dims [2,3,4] // * C is the index vector dimension // * A,B iterate over the updates and indices // // If `update_window_dims` are not the trailing dimensions then updates must be // transposed. // // Eg 2. An update in non-canonical form: // * indices shape(a,b,c) // * updates shape(d,e,f,a,b)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// // For example, // // %shape = tf.Shape(%arg) // %arg: tensor<?x2x3x1xf32> // %height = tf.StridedSlice(%shape, 1, 2, 1) // // In this case %height can be replaced with a constant 2. // // Or, // // %shape = tf.Shape(%arg) // %arg: tensor<?x2x3x1xf32> // %spatial_shape = tf.StridedSlice(%shape, 1, 3, 1) //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)