Searched defs:i_shape (Results 1 – 6 of 6) sorted by relevance
46 const TensorShape i_shape = ctx->InputShape(1); in Compile() local106 const TensorShape i_shape = ctx->InputShape(1); in Compile() local
109 auto i_shape = in Compile() local
168 const TensorShape i_shape = src->tensor_shape(); in validate() local
1893 auto i_shape = in TEST_F() local
2794 def test_shape(i_shape, v_shape, size, expected_size): argument
2991 std::vector<int64_t> i_shape{v_dims[0]}; in TEST_F() local