HomeSort by relevance Sort by last modified time
    Searched refs:expected_diag_shape (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
matrix_set_diag_op.cc 48 TensorShape expected_diag_shape = batch_shape; variable
49 expected_diag_shape.AddDim(min_dim);
50 OP_REQUIRES(context, expected_diag_shape == diag_shape,
  /external/tensorflow/tensorflow/core/kernels/
matrix_set_diag_op.cc 65 TensorShape expected_diag_shape = input_shape; variable
66 expected_diag_shape.RemoveLastDims(2);
67 expected_diag_shape.AddDim(min_dim);
68 OP_REQUIRES(context, expected_diag_shape == diag_shape,

Completed in 68 milliseconds