- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for in_shape (0.24 sec)
-
tensorflow/compiler/jit/xla_tpu_device.cc
TF_RETURN_IF_ERROR( tensorflow::TensorShapeToXLAShape(type, shape, &xla_shape)); ApiConverter::StackHelper<XLA_Shape> se_shape(xla_shape); ApiConverter::StackHelper<XLA_Shape> tpu_shape; StatusHelper status; stream_executor::tpu::ExecutorApiFn()->XlaShapeToTpuShapeRepresentationFn( &se_shape.value, type, use_fast_memory, &tpu_shape.value, status.c_status); if (!status.status().ok()) { return status.status();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composite_functions.mlir
// CHECK-DAG: {{\[\[\[}}-87, -42] // CHECK: %[[quantize:.*]] = "tf.PartitionedCall"(%arg0, %[[in_scale]], %[[in_zp]]) // CHECK-SAME: f = @quantize_i8 // CHECK: %[[conv_quant:.*]] = "tf.PartitionedCall"(%[[quantize]], %[[w_quant]], %[[b_quant]], // CHECK-SAME: %[[in_scale]], %[[in_zp]], %[[w_scale]], %[[w_zp]], // CHECK-SAME: %[[b_scale]], %[[w_zp]], %[[out_scale]], %[[out_zp]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 01:23:21 UTC 2023 - 15.2K bytes - Viewed (0) -
tensorflow/c/BUILD
"@local_xla//xla/tsl/c:tsl_status_internal_headers", ], visibility = [ "//tensorflow/python:__subpackages__", ], ) cc_library( name = "tf_shape", srcs = ["tf_shape.cc"], hdrs = ["tf_shape.h"], copts = tf_copts(), visibility = ["//visibility:public"], deps = [ ":c_api_macros", ":tf_shape_internal", "//tensorflow/core:framework",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/cc/gradients/math_grad.cc
auto x_shape = Shape(scope, x); auto output_shape = Shape(scope, op.output(0)); // Reduce away broadcasted leading dims. auto reduce_x = internal::BroadcastGradientArgs(scope, x_shape, output_shape); auto gx_sum = ReduceSum(scope, gx, /*axis=*/reduce_x.r0, ReduceSum::KeepDims(true)); auto gx_sum_reshape = Reshape(scope, gx_sum, x_shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.7K bytes - Viewed (0) -
src/go/printer/printer.go
_, err = p.output.Write(aNewline) case tabwriter.Escape: _, err = p.output.Write(p.space) p.state = inEscape m = n + 1 // +1: skip tabwriter.Escape default: _, err = p.output.Write(p.space) p.state = inText m = n } case inEscape: if b == tabwriter.Escape { _, err = p.output.Write(data[m:n]) p.resetSpace() } case inText:
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Oct 19 12:02:03 UTC 2023 - 41.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composit_functions_debugging.mlir
// TF-DAG: %[[arg_quantized:.*]] = "tf.PartitionedCall"(%arg0, %[[in_scale]], %[[in_out_zp]]) <{config = "", config_proto = "", executor_type = "", f = @quantize_i8}>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 01:23:21 UTC 2023 - 80.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
// https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/batch-mat-mul int64_t x_row_dim = x_shape[x_shape.size() - 2]; int64_t x_col_dim = x_shape[x_shape.size() - 1]; int64_t y_row_dim = y_shape[y_shape.size() - 2]; int64_t y_col_dim = y_shape[y_shape.size() - 1]; int64_t out_row_dim = output_shape[output_shape.size() - 2]; int64_t out_col_dim = output_shape[output_shape.size() - 1];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_DeleteExecutionContext(ctx); return wrap(func); } TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s) { DCHECK_GE(shape.num_dims, -1); TracingTensorHandle* t; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func)); if (!tracing_ctx) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
) def test_einsum_ptq_model( self, equation: str, ): _, y_shape, bias_shape, x_signature, y_signature = ( self._prepare_sample_einsum_datashapes(equation, use_bias=True) ) model = self._create_einsum_model( self._input_saved_model_path, equation, y_shape, x_signature, y_signature, bias_shape, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composite_functions_xla.mlir
// CHECK-DAG: %[[b0_quantized:.*]] = "tf.Const"() <{value = dense<[120654, 119646]> : tensor<2xi32>}> : () -> tensor<2xi32> // CHECK-DAG: %[[w0_quantized:.*]] = "tf.Const"() <{value = dense<{{\[\[\[\[}}-42, 18 // CHECK-DAG: %[[in_scale:.*]] = "tf.Const"() <{value = dense<0.0039215642> : tensor<f32>}> : () -> tensor<f32> // CHECK-DAG: %[[in_out_zp:.*]] = "tf.Const"() <{value = dense<-128> : tensor<i32>}> : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 08 01:16:10 UTC 2024 - 25.2K bytes - Viewed (0)