Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for shape_shape (0.14 sec)

  1. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      EXPECT_TRUE(clusters.empty());
    }
    
    TEST(XlaCompilationTest, RandomShape) {
      Scope root = Scope::NewRootScope().ExitOnError();
      Output shape_shape = ops::Const(root.WithOpName("shape_shape"), {2}, {1});
      Output shape =
          ops::RandomUniformInt(root.WithOpName("shape"), shape_shape,
                                ops::Const(root.WithOpName("minval"), 1),
                                ops::Const(root.WithOpName("maxval"), 20));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

            if self.has_reshape():
              input_shape = input_tensor.shape
              if len(input_shape) == 3:
                reshape_shape = (input_shape[0], -1, self.bias_size)
              else:
                reshape_shape = (-1, self.bias_size)
    
              out = array_ops.reshape(out, reshape_shape)
    
            if self.has_bias():
              if self.use_biasadd:
                out = nn_ops.bias_add(out, self.bias)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
Back to top