Searched defs:i_shape (Results 1 – 5 of 5) 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
2991 std::vector<int64_t> i_shape{v_dims[0]}; in TEST_F() local