Searched refs:program_shape_output (Results 1 – 2 of 2) sorted by relevance
213 Tensor program_shape_output(DT_STRING, TensorShape({num_cores_per_replica})); in Compute() local217 program_shape_output.vec<tstring>()(i) = program_shape.SerializeAsString(); in Compute()219 ctx->set_output(1, program_shape_output); in Compute()
236 Tensor program_shape_output(DT_STRING, TensorShape({1})); in Compute() local237 program_shape_output.vec<tstring>()(0) = program_shape.SerializeAsString(); in Compute()238 ctx->set_output(1, program_shape_output); in Compute()